pva_algos_detector.h

Go to the documentation of this file.

 
/*
 * Copyright (c) 2024, NVIDIA CORPORATION.  All rights reserved.
 *
 * NVIDIA CORPORATION and its licensors retain all intellectual property
 * and proprietary rights in and to this software, related documentation
 * and any modifications thereto.  Any use, reproduction, disclosure or
 * distribution of this software and related documentation without an express
 * license agreement from NVIDIA CORPORATION is strictly prohibited.
 */
 
#ifndef PVA_ALGOS_DETECTOR
#define PVA_ALGOS_DETECTOR
 
#include <pva_algos_common.h>
 
struct DetectorAlgoContext;
 
typedef struct
{
    uint32_t imageWidth;
    uint32_t imageHeight;
    uint32_t linePitch;
    uint32_t maxFeatureCount;
    uint32_t detectionLevel;
    int32_t maskAdjustmentEnabled;
    int32_t nmsSize;
    float lambda;
    int32_t thresholdFixedPoint;
 
} pvaAlgosDetectorConfig_t;
 
typedef struct DetectorAlgoContext *pvaAlgosDetectorContext_t;
 
typedef struct
{
    uint8_t *image;
    uint32_t *age;
    uint32_t *id;
    uint32_t *outAge;
    uint32_t *outId;
    uint32_t *maxId;
    uint8_t *imageMask;
    int *featureMask;
    uint32_t *status;
    float *loc;
    float *ncc;
    uint32_t *scales;
    int *harrisOutput;
    uint32_t *trackCount;
    uint32_t *detectCount;
    uint32_t *outputFeatureCount;
    float *outLoc;
    uint32_t *outStatus;
    uint32_t *outScales;
    float *outNcc;
    uint32_t *newToOld;
    uint32_t *maxIdD;
    uint32_t *oldFeatureCount;
    bool enableNccScore;
} pvaAlgosDetectorIOBuffer_t;
 
pvaAlgosError_t pvaAlgosDetectorGetContext(pvaAlgosDetectorContext_t *detectorContext);
 
pvaAlgosError_t pvaAlgosDetectorInit(pvaAlgosDetectorContext_t *detectorContext,
                                     pvaAlgosDetectorConfig_t *detectorParams);
 
pvaAlgosError_t pvaAlgosDetectorProcess(pvaAlgosDetectorContext_t *detectorContext,
                                        pvaAlgosDetectorIOBuffer_t *detectorBuffers, cupvaStream_t *instanceStream,
                                        int execTimeout);
 
pvaAlgosError_t pvaAlgosDetectorDeInit(pvaAlgosDetectorContext_t *detectorContext);
 
#endif //PVA_ALGOS_DETECTOR