ActsPlugins::AmbiguityTrackClassifier Class Reference

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

#include </home/runner/work/acts/acts/Plugins/Onnx/include/ActsPlugins/Onnx/AmbiguityTrackClassifier.hpp>

Public Member Functions
	AmbiguityTrackClassifier (const char *modelPath)
	Construct the ambiguity scoring algorithm.
template<Acts::TrackContainerFrontend track_container_t>
std::vector< std::vector< float > >	inferScores (std::unordered_map< std::size_t, std::vector< std::size_t > > &clusters, const track_container_t &tracks) const
	Compute a score for each track to be used in the track selection.
std::vector< std::size_t >	trackSelection (std::unordered_map< std::size_t, std::vector< std::size_t > > &clusters, std::vector< std::vector< float > > &outputTensor) const
	Select the track associated with each cluster based on the score vector.

Detailed Description

Onnx model implementation for track scoring and selection.

Constructor & Destructor Documentation

AmbiguityTrackClassifier()

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

explicit

Construct the ambiguity scoring algorithm.

Parameters

modelPath

path to the model file

Member Function Documentation

inferScores()

template<Acts::TrackContainerFrontend track_container_t>

std::vector< std::vector< float > > ActsPlugins::AmbiguityTrackClassifier::inferScores	(	std::unordered_map< std::size_t, std::vector< std::size_t > > &	clusters,
		const track_container_t &	tracks ) const

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.

trackSelection()

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

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