Skip to content

Add Graph Neural Network-based track finding

Xiangyang Ju requested to merge xju/athena:mr_exatrkx_onnx into main

This MR introduces one new package InnerDetector/InDetGNNTracking that contains three components:

  1. SeedFitterTool This tool is to estimate the track parameter for track seeds. It is used in the SiSPGNNTrackMaker algorithm.
  2. SiGNNTrackFinderTool This tool implements the Graph Neural Network-based track-finding algorithm. The current implementation uses the metric learning method for graph construction. The Neural Networks, including the GNN, are executed via the ONNX Runtime.
  3. SiSPGNNTrackMaker This algorithm is the main algorithm that takes a list of spacepoints and returns a track container.
Edited by Vakhtang Tsulaia

Merge request reports

Loading