Skip to content

Add NNTrackTool functionality to FPGATrackSim

This is adding functionality to the FPGATrackSimNNTrackTool class to read a NN from an ONNX file and apply it for fake track removal in FPGATrackSim. For this a method is added to the FPGATrackSimNNMap class to pass the path to that ONNX file via the Mapping service. An object of that class is then created in FPGATrackSimMappingSvc to be able to then access the maps stored in there in a method in the MappingSvc, which in turn can then be accessed in the NNTrackTool.

For the actual application of the NN in the tracking setup, the OnnxRuntimeBase class is added.

Furthermore, the call to the mapping service in the recently updated SecondStage algorithm is modified slightly to fit the setup.

@sabidi @jahreda @brosser

Merge request reports

Loading