Python/src/Onnx.cpp

0001 // This file is part of the ACTS project.
0002 //
0003 // Copyright (C) 2016 CERN for the benefit of the ACTS project
0004 //
0005 // This Source Code Form is subject to the terms of the Mozilla Public
0006 // License, v. 2.0. If a copy of the MPL was not distributed with this
0007 // file, You can obtain one at https://mozilla.org/MPL/2.0/.
0008
0009 #include "ActsExamples/TrackFindingML/AmbiguityResolutionMLAlgorithm.hpp"
0010 #include "ActsExamples/TrackFindingML/SeedFilterMLAlgorithm.hpp"
0011 #include "ActsPython/Utilities/Helpers.hpp"
0012 #include "ActsPython/Utilities/Macros.hpp"
0013
0014 #include <pybind11/pybind11.h>
0015 #include <pybind11/stl.h>
0016
0017 namespace py = pybind11;
0018
0019 using namespace ActsExamples;
0020 using namespace Acts;
0021
0022 namespace ActsPython {
0023
0024 void addOnnx(Context& ctx) {
0025   auto [m, mex] = ctx.get("main", "examples");
0026   auto onnx = mex.def_submodule("_onnx");
0027   ctx.modules["onnx"] = onnx;
0028
0029   ACTS_PYTHON_DECLARE_ALGORITHM(
0030       AmbiguityResolutionMLAlgorithm, onnx, "AmbiguityResolutionMLAlgorithm",
0031       inputTracks, inputDuplicateNN, outputTracks, nMeasurementsMin);
0032
0033   ACTS_PYTHON_DECLARE_ALGORITHM(SeedFilterMLAlgorithm, onnx,
0034                                 "SeedFilterMLAlgorithm", inputTrackParameters,
0035                                 inputSimSeeds, inputSeedFilterNN,
0036                                 outputTrackParameters, outputSimSeeds,
0037                                 epsilonDBScan, minPointsDBScan, minSeedScore);
0038 }
0039 }  // namespace ActsPython