Acts::AmbiguityTrackClassifier Class Reference

Onnx model implementation for track scoring and selection. More...

#include <acts/blob/sPHENIX/Plugins/Onnx/include/Acts/Plugins/Onnx/AmbiguityTrackClassifier.hpp>

Collaboration diagram for Acts::AmbiguityTrackClassifier:

Public Member Functions
	AmbiguityTrackClassifier (const char *modelPath)
template<typename track_container_t , typename traj_t , template< typename > class holder_t>
std::vector< std::vector< float > >	inferScores (std::unordered_map< int, std::vector< int >> &clusters, const Acts::TrackContainer< track_container_t, traj_t, holder_t > &tracks) const
std::vector< int >	trackSelection (std::unordered_map< int, std::vector< int >> &clusters, std::vector< std::vector< float >> &outputTensor) const
template<typename track_container_t , typename traj_t , template< typename > class holder_t>
std::vector< int >	solveAmbuguity (std::unordered_map< int, std::vector< int >> &clusters, const Acts::TrackContainer< track_container_t, traj_t, holder_t > &tracks) const

Private Attributes
Ort::Env	m_env
Acts::OnnxRuntimeBase	m_duplicateClassifier

Detailed Description

Onnx model implementation for track scoring and selection.

Definition at line 25 of file AmbiguityTrackClassifier.hpp.

View newest version in sPHENIX GitHub at line 25 of file AmbiguityTrackClassifier.hpp

Constructor & Destructor Documentation

Acts::AmbiguityTrackClassifier::AmbiguityTrackClassifier ( const char *  modelPath )

inline

Construct the ambiguity scoring algorithm.

Parameters

modelPath

path to the model file

Definition at line 30 of file AmbiguityTrackClassifier.hpp.

View newest version in sPHENIX GitHub at line 30 of file AmbiguityTrackClassifier.hpp

Member Function Documentation

template<typename track_container_t , typename traj_t , template< typename > class holder_t>

std::vector<std::vector<float> > Acts::AmbiguityTrackClassifier::inferScores ( std::unordered_map< int, std::vector< int >> &  clusters, const Acts::TrackContainer< track_container_t, traj_t, holder_t > &  tracks  ) const

inline

Compute a score for each track to be used in the track selection

Parameters

clusters	is a map of clusters, each cluster correspond to a vector of track ID
tracks	is the input track container

Returns

a vector of vector of track score. Due to the architecture of the network each track only have a size 1 score vector.

Definition at line 41 of file AmbiguityTrackClassifier.hpp.

View newest version in sPHENIX GitHub at line 41 of file AmbiguityTrackClassifier.hpp

References _, clusters, Acts::VectorHelpers::eta(), m_duplicateClassifier, Acts::MultiTrajectoryHelpers::TrajectoryState::nStates, Acts::VectorHelpers::phi(), Acts::OnnxRuntimeBase::runONNXInference(), trackID, and Acts::MultiTrajectoryHelpers::trajectoryState().

Referenced by solveAmbuguity().

Here is the call graph for this function:

Here is the caller graph for this function:

template<typename track_container_t , typename traj_t , template< typename > class holder_t>

std::vector<int> Acts::AmbiguityTrackClassifier::solveAmbuguity ( std::unordered_map< int, std::vector< int >> &  clusters, const Acts::TrackContainer< track_container_t, traj_t, holder_t > &  tracks  ) const

inline

Select the track associated with each cluster

Parameters

clusters	is a map of clusters, each cluster correspond to a vector of track ID
tracks	is the input track container

Returns

a vector of trackID corresponding tho the good tracks

Definition at line 112 of file AmbiguityTrackClassifier.hpp.

View newest version in sPHENIX GitHub at line 112 of file AmbiguityTrackClassifier.hpp

References clusters, inferScores(), and trackSelection().

Here is the call graph for this function:

std::vector<int> Acts::AmbiguityTrackClassifier::trackSelection ( std::unordered_map< int, std::vector< int >> &  clusters, std::vector< std::vector< float >> &  outputTensor  ) const

inline

Select the track associated with each cluster based on the score vector

Parameters

clusters	is a map of clusters, each cluster correspond to a vector of track ID
outputTensor	is the score vector obtained from inferScores.

Returns

a vector of trackID corresponding tho the good tracks

Definition at line 83 of file AmbiguityTrackClassifier.hpp.

View newest version in sPHENIX GitHub at line 83 of file AmbiguityTrackClassifier.hpp

References clusters.

Referenced by solveAmbuguity().

Here is the caller graph for this function:

Member Data Documentation

Acts::OnnxRuntimeBase Acts::AmbiguityTrackClassifier::m_duplicateClassifier

private

Definition at line 127 of file AmbiguityTrackClassifier.hpp.

View newest version in sPHENIX GitHub at line 127 of file AmbiguityTrackClassifier.hpp

Referenced by inferScores().

Ort::Env Acts::AmbiguityTrackClassifier::m_env

private

Definition at line 125 of file AmbiguityTrackClassifier.hpp.

View newest version in sPHENIX GitHub at line 125 of file AmbiguityTrackClassifier.hpp

---

The documentation for this class was generated from the following file:

• acts/blob/sPHENIX/Plugins/Onnx/include/Acts/Plugins/Onnx/AmbiguityTrackClassifier.hpp