Detailed Description

Stores and updates per-target ReID feature histories.

Definition at line 28 of file sources/libs/nvmultiobjecttracker/include/modules/ReID/ReIDFeatureGallery.hpp.

Public Member Functions
	ReIDFeatureGallery ()
	Construct a ReIDFeatureGallery instance. More...

	~ReIDFeatureGallery ()
	Release resources owned by the ReIDFeatureGallery instance. More...

void	initialize (const int reidFeatureSize, const int reidHistorySize, const uint32_t maxStreams, const int maxTargetsPerStream, const int debugVerboseLevel)
	Allocate feature-gallery storage and initialize stream partitions. More...

int	getReidFeatureIndGlobal (const NvMOTStreamId videoStreamId, uint targetInd)
	For a new reid feature to add, determine its position in feature gallery and return the global index of that position. More...

int	getReidFeatureOffset (const NvMOTStreamId videoStreamId) const
	Get the global-gallery offset of a stream's partition. More...

void	remove (const NvMOTStreamId videoStreamId, const std::vector< int > &reidFeatureInds)
	Remove reid feature of given indices. More...

void	removeStream (const NvMOTStreamId videoStreamId)
	Remove a stream's all reid features. More...

bool	updateReIDFeature (const NvMOTStreamId videoStreamId, const DetectedObjInfo *detObj, const std::shared_ptr< NvTrackedObject > target)
	Update reid feature gallery and each target's record. More...

void	removeStreamReIDFeature (const NvMOTStreamId videoStreamId)
	Remove reid feature gallery and each target's record. More...

void	removeTrackerReIDFeature (const NvMOTStreamId videoStreamId, const std::shared_ptr< NvTrackedObject > target)
	Remove a target's reid features from the gallery. More...

void	fuseReidFeatureInd (const NvMOTStreamId videoStreamId, uint targetIndKept, uint targetIndFused)
	Merge fused targets' reid features into its peer. More...

const int	getLastInd (const NvMOTStreamId streamId) const
	Get the last feature gallery index in use for a stream. More...

const float *	getReIDFeaturePtrs (const NvMOTStreamId streamId) const
	Get the device pointer to a stream's feature gallery partition. More...

float *	getFirstReIDFeaturePtrs ()
	Get the start of the complete feature gallery. More...

const int	getReIDHistorySize () const
	Get the configured per-target feature history size. More...

const std::vector< int > &	getTargetReIDFeatureInds (uint targetInd) const
	Get the gallery feature indices retained for a target. More...

const int	getTargetReIDFeatureNextInd (uint targetInd) const
	Get the next circular-history slot for a target. More...

void	resetTargetReIDFeatureInds (uint targetInd)
	Reset the feature indices retained for a target. More...

void	updateTargetReIDFeatureInd (uint targetInd, int featureInd)
	Record a feature index for a target. More...

void	printFeatureInds (uint targetInd) const
	Print the feature indices retained for a target. More...

	ReIDFeatureGallery ()
	Construct a ReIDFeatureGallery instance. More...

	~ReIDFeatureGallery ()
	Release resources owned by the ReIDFeatureGallery instance. More...

void	initialize (const int reidFeatureSize, const int reidHistorySize, const uint32_t maxStreams, const int maxTargetsPerStream, const int debugVerboseLevel)
	Allocate feature-gallery storage and initialize stream partitions. More...

int	getReidFeatureIndGlobal (const NvMOTStreamId videoStreamId, uint targetInd)
	For a new reid feature to add, determine its position in feature gallery and return the global index of that position. More...

int	getReidFeatureOffset (const NvMOTStreamId videoStreamId) const
	Get the global-gallery offset of a stream's partition. More...

void	remove (const NvMOTStreamId videoStreamId, const std::vector< int > &reidFeatureInds)
	Remove reid feature of given indices. More...

void	removeStream (const NvMOTStreamId videoStreamId)
	Remove a stream's all reid features. More...

bool	updateReIDFeature (const NvMOTStreamId videoStreamId, const DetectedObjInfo *detObj, const std::shared_ptr< NvTrackedObject > target)
	Update reid feature gallery and each target's record. More...

void	removeStreamReIDFeature (const NvMOTStreamId videoStreamId)
	Remove reid feature gallery and each target's record. More...

void	removeTrackerReIDFeature (const NvMOTStreamId videoStreamId, const std::shared_ptr< NvTrackedObject > target)
	Remove a target's reid features from the gallery. More...

void	fuseReidFeatureInd (const NvMOTStreamId videoStreamId, uint targetIndKept, uint targetIndFused)
	Merge fused targets' reid features into its peer. More...

const int	getLastInd (const NvMOTStreamId streamId) const
	Get the last feature gallery index in use for a stream. More...

const float *	getReIDFeaturePtrs (const NvMOTStreamId streamId) const
	Get the device pointer to a stream's feature gallery partition. More...

float *	getFirstReIDFeaturePtrs ()
	Get the start of the complete feature gallery. More...

const int	getReIDHistorySize () const
	Get the configured per-target feature history size. More...

const std::vector< int > &	getTargetReIDFeatureInds (uint targetInd) const
	Get the gallery feature indices retained for a target. More...

const int	getTargetReIDFeatureNextInd (uint targetInd) const
	Get the next circular-history slot for a target. More...

void	resetTargetReIDFeatureInds (uint targetInd)
	Reset the feature indices retained for a target. More...

void	updateTargetReIDFeatureInd (uint targetInd, int featureInd)
	Record a feature index for a target. More...

void	printFeatureInds (uint targetInd) const
	Print the feature indices retained for a target. More...

Protected Attributes
int	m_debugVerboseLevel
	Debug message verbose mode level. More...

int	m_maxTargetsPerStream
	Max targets per stream. More...

int	m_reidHistorySize
	Max number of reid features per object. More...

int	m_reidFeatureSize
	Size of each reid feature. More...

std::map< NvMOTStreamId, int >	m_reidFeatureOffsets
	Reid feature gallery indices being used by each stream, {streamID, ptr}. More...

std::deque< int >	m_reidFeatureOffsetsInPool
	Reid feature pointers in pool when unused. More...

std::map< NvMOTStreamId, std::set< int > >	m_reidFeatureIndsInPool
	Feature gallery indices in pool. More...

std::map< NvMOTStreamId, int >	m_lastInds
	Index of last feature gallery being used in each stream. More...

std::map< uint, std::vector< int > >	m_reidFeatureIndsTargets
	Each target's reid feature indices. More...

std::map< uint, uint >	m_reidFeatureNextIndTargets
	Each target's next circular-history slot for a reid feature. More...

float *	m_reidFeatureGallery_d

float *	m_reidFeatureGallery_h
	CUDA array saving target features in past frames. More...

cudaStream_t	m_cuStream
	CUDA stream. More...

bool	m_isInit
	If initialized. More...

Constructor & Destructor Documentation

◆ ReIDFeatureGallery() [1/2]

ReIDFeatureGallery::ReIDFeatureGallery ( )

Construct a ReIDFeatureGallery instance.

◆ ~ReIDFeatureGallery() [1/2]

ReIDFeatureGallery::~ReIDFeatureGallery ( )

Release resources owned by the ReIDFeatureGallery instance.

◆ ReIDFeatureGallery() [2/2]

ReIDFeatureGallery::ReIDFeatureGallery ( )

Construct a ReIDFeatureGallery instance.

◆ ~ReIDFeatureGallery() [2/2]

ReIDFeatureGallery::~ReIDFeatureGallery ( )

Release resources owned by the ReIDFeatureGallery instance.

Member Function Documentation

◆ fuseReidFeatureInd() [1/2]

void ReIDFeatureGallery::fuseReidFeatureInd	(	const NvMOTStreamId	videoStreamId,
		uint	targetIndKept,
		uint	targetIndFused
	)

Merge fused targets' reid features into its peer.

Parameters

[in]	videoStreamId	Video stream identifier.
[in]	targetIndKept	Index of the target that keeps the merged features.
[in]	targetIndFused	Index of the target whose features are merged into the kept target.

◆ fuseReidFeatureInd() [2/2]

void ReIDFeatureGallery::fuseReidFeatureInd	(	const NvMOTStreamId	videoStreamId,
		uint	targetIndKept,
		uint	targetIndFused
	)

Merge fused targets' reid features into its peer.

Parameters

[in]	videoStreamId	Video stream identifier.
[in]	targetIndKept	Index of the target that keeps the merged features.
[in]	targetIndFused	Index of the target whose features are merged into the kept target.

◆ getFirstReIDFeaturePtrs() [1/2]

float* ReIDFeatureGallery::getFirstReIDFeaturePtrs ( )

inline

Get the start of the complete feature gallery.

Definition at line 128 of file sources/libs/nvmultiobjecttracker/include/modules/ReID/ReIDFeatureGallery.hpp.

References m_reidFeatureGallery_d.

◆ getFirstReIDFeaturePtrs() [2/2]

float* ReIDFeatureGallery::getFirstReIDFeaturePtrs ( )

inline

Get the start of the complete feature gallery.

Definition at line 128 of file 9.2/sources/libs/nvmultiobjecttracker/include/modules/ReID/ReIDFeatureGallery.hpp.

References m_reidFeatureGallery_d.

◆ getLastInd() [1/2]

const int ReIDFeatureGallery::getLastInd ( const NvMOTStreamId streamId ) const

Get the last feature gallery index in use for a stream.

Parameters

[in] streamId Video stream identifier.

Returns: The last index in use, or -1 when the stream has no gallery partition.

◆ getLastInd() [2/2]

const int ReIDFeatureGallery::getLastInd ( const NvMOTStreamId streamId ) const

Get the last feature gallery index in use for a stream.

Parameters

[in] streamId Video stream identifier.

Returns: The last index in use, or -1 when the stream has no gallery partition.

◆ getReidFeatureIndGlobal() [1/2]

int ReIDFeatureGallery::getReidFeatureIndGlobal	(	const NvMOTStreamId	videoStreamId,
		uint	targetInd
	)

For a new reid feature to add, determine its position in feature gallery and return the global index of that position.

Parameters

[in]	videoStreamId	Video stream identifier.
[in]	targetInd	Index of the target that receives the feature.

Returns: Global index in the feature gallery, or -1 when the feature cannot be added.

◆ getReidFeatureIndGlobal() [2/2]

int ReIDFeatureGallery::getReidFeatureIndGlobal	(	const NvMOTStreamId	videoStreamId,
		uint	targetInd
	)

For a new reid feature to add, determine its position in feature gallery and return the global index of that position.

Parameters

[in]	videoStreamId	Video stream identifier.
[in]	targetInd	Index of the target that receives the feature.

Returns: Global index in the feature gallery, or -1 when the feature cannot be added.

◆ getReidFeatureOffset() [1/2]

int ReIDFeatureGallery::getReidFeatureOffset ( const NvMOTStreamId videoStreamId ) const

inline

Get the global-gallery offset of a stream's partition.

Parameters

[in] videoStreamId Video stream id.

Returns: The stream's offset into the global gallery, or -1 when the stream has no gallery partition.

Definition at line 63 of file 9.2/sources/libs/nvmultiobjecttracker/include/modules/ReID/ReIDFeatureGallery.hpp.

References m_reidFeatureOffsets.

◆ getReidFeatureOffset() [2/2]

int ReIDFeatureGallery::getReidFeatureOffset ( const NvMOTStreamId videoStreamId ) const

inline

Get the global-gallery offset of a stream's partition.

Parameters

[in] videoStreamId Video stream id.

Returns: The stream's offset into the global gallery, or -1 when the stream has no gallery partition.

Definition at line 63 of file sources/libs/nvmultiobjecttracker/include/modules/ReID/ReIDFeatureGallery.hpp.

References m_reidFeatureOffsets.

◆ getReIDFeaturePtrs() [1/2]

const float* ReIDFeatureGallery::getReIDFeaturePtrs ( const NvMOTStreamId streamId ) const

Get the device pointer to a stream's feature gallery partition.

Parameters

[in] streamId Video stream identifier.

Returns: The device pointer to the stream's features, or nullptr when the stream has no gallery partition.

◆ getReIDFeaturePtrs() [2/2]

const float* ReIDFeatureGallery::getReIDFeaturePtrs ( const NvMOTStreamId streamId ) const

Get the device pointer to a stream's feature gallery partition.

Parameters

[in] streamId Video stream identifier.

Returns: The device pointer to the stream's features, or nullptr when the stream has no gallery partition.

◆ getReIDHistorySize() [1/2]

const int ReIDFeatureGallery::getReIDHistorySize ( ) const

inline

Get the configured per-target feature history size.

Definition at line 131 of file sources/libs/nvmultiobjecttracker/include/modules/ReID/ReIDFeatureGallery.hpp.

References m_reidHistorySize.

◆ getReIDHistorySize() [2/2]

const int ReIDFeatureGallery::getReIDHistorySize ( ) const

inline

Get the configured per-target feature history size.

Definition at line 131 of file 9.2/sources/libs/nvmultiobjecttracker/include/modules/ReID/ReIDFeatureGallery.hpp.

References m_reidHistorySize.

◆ getTargetReIDFeatureInds() [1/2]

const std::vector<int>& ReIDFeatureGallery::getTargetReIDFeatureInds ( uint targetInd ) const

inline

Get the gallery feature indices retained for a target.

Parameters

[in] targetInd Target index.

Definition at line 137 of file 9.2/sources/libs/nvmultiobjecttracker/include/modules/ReID/ReIDFeatureGallery.hpp.

References m_reidFeatureIndsTargets.

◆ getTargetReIDFeatureInds() [2/2]

const std::vector<int>& ReIDFeatureGallery::getTargetReIDFeatureInds ( uint targetInd ) const

inline

Get the gallery feature indices retained for a target.

Parameters

[in] targetInd Target index.

Definition at line 137 of file sources/libs/nvmultiobjecttracker/include/modules/ReID/ReIDFeatureGallery.hpp.

References m_reidFeatureIndsTargets.

◆ getTargetReIDFeatureNextInd() [1/2]

const int ReIDFeatureGallery::getTargetReIDFeatureNextInd ( uint targetInd ) const

inline

Get the next circular-history slot for a target.

Parameters

[in] targetInd Target index.

Definition at line 146 of file 9.2/sources/libs/nvmultiobjecttracker/include/modules/ReID/ReIDFeatureGallery.hpp.

References m_reidFeatureNextIndTargets.

◆ getTargetReIDFeatureNextInd() [2/2]

const int ReIDFeatureGallery::getTargetReIDFeatureNextInd ( uint targetInd ) const

inline

Get the next circular-history slot for a target.

Parameters

[in] targetInd Target index.

Definition at line 146 of file sources/libs/nvmultiobjecttracker/include/modules/ReID/ReIDFeatureGallery.hpp.

References m_reidFeatureNextIndTargets.

◆ initialize() [1/2]

void ReIDFeatureGallery::initialize	(	const int	reidFeatureSize,
		const int	reidHistorySize,
		const uint32_t	maxStreams,
		const int	maxTargetsPerStream,
		const int	debugVerboseLevel
	)

Allocate feature-gallery storage and initialize stream partitions.

Parameters

[in]	reidFeatureSize	Number of values in each ReID feature.
[in]	reidHistorySize	Maximum number of ReID features per object.
[in]	maxStreams	Maximum number of streams.
[in]	maxTargetsPerStream	Maximum number of targets to track in each stream.
[in]	debugVerboseLevel	Debug verbosity level.

◆ initialize() [2/2]

void ReIDFeatureGallery::initialize	(	const int	reidFeatureSize,
		const int	reidHistorySize,
		const uint32_t	maxStreams,
		const int	maxTargetsPerStream,
		const int	debugVerboseLevel
	)

Allocate feature-gallery storage and initialize stream partitions.

Parameters

[in]	reidFeatureSize	Number of values in each ReID feature.
[in]	reidHistorySize	Maximum number of ReID features per object.
[in]	maxStreams	Maximum number of streams.
[in]	maxTargetsPerStream	Maximum number of targets to track in each stream.
[in]	debugVerboseLevel	Debug verbosity level.

◆ printFeatureInds() [1/2]

void ReIDFeatureGallery::printFeatureInds ( uint targetInd ) const

inline

Print the feature indices retained for a target.

Parameters

[in] targetInd Target index.

Definition at line 180 of file sources/libs/nvmultiobjecttracker/include/modules/ReID/ReIDFeatureGallery.hpp.

References m_reidFeatureIndsTargets.

◆ printFeatureInds() [2/2]

void ReIDFeatureGallery::printFeatureInds ( uint targetInd ) const

inline

Print the feature indices retained for a target.

Parameters

[in] targetInd Target index.

Definition at line 180 of file 9.2/sources/libs/nvmultiobjecttracker/include/modules/ReID/ReIDFeatureGallery.hpp.

References m_reidFeatureIndsTargets.

◆ remove() [1/2]

void ReIDFeatureGallery::remove	(	const NvMOTStreamId	videoStreamId,
		const std::vector< int > &	reidFeatureInds
	)

Remove reid feature of given indices.

Parameters

[in]	videoStreamId	Video stream identifier.
[in]	reidFeatureInds	Indices of the features to remove.

◆ remove() [2/2]

void ReIDFeatureGallery::remove	(	const NvMOTStreamId	videoStreamId,
		const std::vector< int > &	reidFeatureInds
	)

Remove reid feature of given indices.

Parameters

[in]	videoStreamId	Video stream identifier.
[in]	reidFeatureInds	Indices of the features to remove.

◆ removeStream() [1/2]

void ReIDFeatureGallery::removeStream ( const NvMOTStreamId videoStreamId )

Remove a stream's all reid features.

Parameters

[in] videoStreamId Video stream id.

◆ removeStream() [2/2]

void ReIDFeatureGallery::removeStream ( const NvMOTStreamId videoStreamId )

Remove a stream's all reid features.

Parameters

[in] videoStreamId Video stream id.

◆ removeStreamReIDFeature() [1/2]

void ReIDFeatureGallery::removeStreamReIDFeature ( const NvMOTStreamId videoStreamId )

Remove reid feature gallery and each target's record.

Parameters

[in] videoStreamId Video stream id.

◆ removeStreamReIDFeature() [2/2]

void ReIDFeatureGallery::removeStreamReIDFeature ( const NvMOTStreamId videoStreamId )

Remove reid feature gallery and each target's record.

Parameters

[in] videoStreamId Video stream id.

◆ removeTrackerReIDFeature() [1/2]

void ReIDFeatureGallery::removeTrackerReIDFeature	(	const NvMOTStreamId	videoStreamId,
		const std::shared_ptr< NvTrackedObject >	target
	)

Remove a target's reid features from the gallery.

Parameters

[in]	videoStreamId	Video stream id.
[in]	target	Tracker associated with the detection.

◆ removeTrackerReIDFeature() [2/2]

void ReIDFeatureGallery::removeTrackerReIDFeature	(	const NvMOTStreamId	videoStreamId,
		const std::shared_ptr< NvTrackedObject >	target
	)

Remove a target's reid features from the gallery.

Parameters

[in]	videoStreamId	Video stream id.
[in]	target	Tracker associated with the detection.

◆ resetTargetReIDFeatureInds() [1/2]

void ReIDFeatureGallery::resetTargetReIDFeatureInds ( uint targetInd )

inline

Reset the feature indices retained for a target.

Parameters

[in] targetInd Target index.

Definition at line 155 of file 9.2/sources/libs/nvmultiobjecttracker/include/modules/ReID/ReIDFeatureGallery.hpp.

References m_reidFeatureIndsTargets, and m_reidFeatureNextIndTargets.

◆ resetTargetReIDFeatureInds() [2/2]

void ReIDFeatureGallery::resetTargetReIDFeatureInds ( uint targetInd )

inline

Reset the feature indices retained for a target.

Parameters

[in] targetInd Target index.

Definition at line 155 of file sources/libs/nvmultiobjecttracker/include/modules/ReID/ReIDFeatureGallery.hpp.

References m_reidFeatureIndsTargets, and m_reidFeatureNextIndTargets.

◆ updateReIDFeature() [1/2]

bool ReIDFeatureGallery::updateReIDFeature	(	const NvMOTStreamId	videoStreamId,
		const DetectedObjInfo *	detObj,
		const std::shared_ptr< NvTrackedObject >	target
	)

Update reid feature gallery and each target's record.

Parameters

[in]	videoStreamId	Video stream id.
[in]	detObj	Detection that provides the new feature.
[in]	target	The target associated with this detection.

◆ updateReIDFeature() [2/2]

bool ReIDFeatureGallery::updateReIDFeature	(	const NvMOTStreamId	videoStreamId,
		const DetectedObjInfo *	detObj,
		const std::shared_ptr< NvTrackedObject >	target
	)