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 "Acts/Plugins/Python/Utilities.hpp"
0010 #include "ActsExamples/TrackFindingML/AmbiguityResolutionMLAlgorithm.hpp"
0011 #include "ActsExamples/TrackFindingML/SeedFilterMLAlgorithm.hpp"
0012
0013 #include <pybind11/pybind11.h>
0014 #include <pybind11/stl.h>
0015
0016 namespace py = pybind11;
0017
0018 using namespace ActsExamples;
0019 using namespace Acts;
0020
0021 namespace Acts::Python {
0022
0023 void addOnnx(Context& ctx) {
0024   auto [m, mex] = ctx.get("main", "examples");
0025   auto onnx = mex.def_submodule("_onnx");
0026   ctx.modules["onnx"] = onnx;
0027
0028   ACTS_PYTHON_DECLARE_ALGORITHM(ActsExamples::AmbiguityResolutionMLAlgorithm,
0029                                 onnx, "AmbiguityResolutionMLAlgorithm",
0030                                 inputTracks, inputDuplicateNN, outputTracks,
0031                                 nMeasurementsMin);
0032
0033   ACTS_PYTHON_DECLARE_ALGORITHM(ActsExamples::SeedFilterMLAlgorithm, onnx,
0034                                 "SeedFilterMLAlgorithm", inputTrackParameters,
0035                                 inputSimSeeds, inputSeedFilterNN,
0036                                 outputTrackParameters, outputSimSeeds,
0037                                 epsilonDBScan, minPointsDBScan, minSeedScore);
0038 }
0039 }  // namespace Acts::Python