NVIDIA DeepStream SDK API Reference

9.2 Release
sources/libs/nvmultiobjecttracker/include/modules/cuDCFv2/cuDCF.h File Reference

Go to the source code of this file.

Data Structures

struct  detectorBbox_t
 Detector bbox in correlation-response coordinates. More...
 

Namespaces

 cuDCF
 

Typedefs

typedef float ftype1
 
typedef float2 ftype2
 

Functions

void cuDCF::getDetectorBboxCorrelationResponses (const cudaStream_t &stream, const int num_of_targets, const int maxTargetsPerStream, const detectorBbox_t *detectedPts, const uint *detectedPts_size, const int detCorrRespSearchWindowSize, const float *corr_responses, const float *maxCorrResponse_ma, const int img_element_size, const int img_width, float *confidenceAtDetectedPt, float *corrRespAtDetectedPt)
 Read the correlation response near each detector bbox and write the confidence and response. More...
 
void cuDCF::applyFilter2D_U16 (const cudaStream_t &stream, const uint16_t *ftrs, const float *masking_window, const uint srcWidth, const uint srcPitch, const uint srcHeight, const uint2 outBatchingGrid, const uint num_of_channels, const uint num_of_targets, const bool useColorNames, const uint num_of_colorNames_channels, float *ftrs_masked)
 Multiply a 2D high-precision feature by a masking window elementwise. More...
 
void cuDCF::applyFilter2D_U8 (const cudaStream_t &stream, const uchar *ftrs, const float *masking_window, const uint srcWidth, const uint srcPitch, const uint srcHeight, const uint2 outBatchingGrid, const uint num_of_channels, const uint num_of_targets, const bool useColorNames, const uint num_of_colorNames_channels, float *ftrs_masked)
 Multiply a 2D low-precision feature by a masking window elementwise. More...
 
void cuDCF::createDcfFilter (const cudaStream_t &stream, const cufftHandle &cufft_handle_C2R, const cufftHandle &cufft_handle_R2C, const cuComplex *ftrs_F, const cuComplex *desiredResp_F, const float *masking_window_model, const int img_width, const int img_element_size, const int batch_element_size, const float lambda, const int num_of_channels, const int num_of_targets, cuComplex *work_F, float *work_I, const uint32_t *tracker_age, const bool *trackerDoProcessing_d, const float learn_rate, cuComplex *learned_filter_F, cuComplex *result_filter_F)
 Create a DCF filter and update the learned filter with an exponential moving average. More...
 
void cuDCF::createSegmentationFilterMask (const cudaStream_t &stream, const float *src_mask_frame_batch, const int src_frame_width, const int src_frame_element_size, const int feat_width, const int feat_element_size, const int batch_element_size, const int num_of_targets, const int num_of_channels, const float4 *src_rois, const bool *trackerDoProcessing_d, float *dst_mask_patch_batch)
 Crop and rescale each target's segmentation mask to the visual tracker region size. More...
 
void cuDCF::computeChannelWeights (const cudaStream_t &stream, const cufftHandle &cufft_handle_many, const cuComplex *ftrs_F, const int img_element_size, const int img_width, const int batch_element_size, const int batch_size, const int num_of_channels, const int num_of_targets, const cuComplex *new_filter_F, cuComplex *work_F, float *work_I, const uint32_t *tracker_age, const bool *trackerDoProcessing_d, const float weights_lr, float *channel_weights, float *weight_sum)
 Compute and update the channel weights via exponential moving average. More...
 
void cuDCF::calcuateExpectedPeakCorrelation (const cudaStream_t &stream, const int num_of_channels, const int num_of_targets, const float learn_rate, const float *channel_weights, const uint32_t *tracker_age, const bool *trackerDoProcessing_d, float *maxCorrResponse_ma)
 Compute the moving average of the max correlation response. More...
 
void cuDCF::calculateResponse (const cudaStream_t &stream, const cufftHandle &cufft_handle_many_small, const int img_width, const int img_element_size, const int batch_element_size, const int num_of_channels, const int num_of_targets, const cuComplex *ftrs_F, const cuComplex *result_filter_F, const float *channel_weights, cuComplex *work_F, float *corr_responses, float *maxResp, float2 *targetDisplacement)
 Given the Fourier-transformed extracted features and the learned filters, compute the correlation response. More...
 
void cuDCF::calculateTrackingConfidence (const cudaStream_t &stream, const int num_of_targets, const float *maxCorrResponse_ma, const float *maxCorrResponse, float *trackingConfidence)
 Calculate tracking confidence from moving-average and instantaneous peak correlation responses. More...
 

Typedef Documentation

◆ ftype1

◆ ftype2