pva_algos_object_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_OBJECT_DETECTOR
#define PVA_ALGOS_OBJECT_DETECTOR
 
#include <pva_algos_common.h>
 
struct ObjectDetectorContext;
 
typedef enum pvaAlgosObjectClass
{
    PERSON,
    OBJECT_CLASS_MAX
} pvaAlgosObjectClass;
 
typedef struct
{
    uint32_t imageWidth;
    uint32_t imageHeight;
    uint32_t linePitch;
    uint32_t maxBbox;
    uint32_t num_classes;
    pvaAlgosObjectClass *class_names;
    float conf_thresh;
    float iou_thresh;
} pvaAlgosObjectDetectorConfig_t;
 
typedef struct
{
    uint8_t *image;
    float *bboxesScores;
    uint32_t *numBbox;
} pvaAlgosObjectDetectorIOBuffer_t;
 
typedef struct ObjectDetectorContext *pvaAlgosObjectDetectorContext_t;
pvaAlgosError_t pvaAlgosObjectDetectorGetContext(pvaAlgosObjectDetectorContext_t *ObjectDetectorContext);
 
pvaAlgosError_t pvaAlgosObjectDetectorInit(pvaAlgosObjectDetectorContext_t *ObjectDetectorContext,
                                           pvaAlgosObjectDetectorConfig_t *ObjectDetectorParams);
 
pvaAlgosError_t pvaAlgosObjectDetectorProcess(pvaAlgosObjectDetectorContext_t *ObjectDetectorContext,
                                              pvaAlgosObjectDetectorIOBuffer_t *ObjectDetectorBuffers,
                                              cupvaStream_t *instanceStream, int execTimeout);
 
pvaAlgosError_t pvaAlgosObjectDetectorDeInit(pvaAlgosObjectDetectorContext_t *ObjectDetectorContext);
 
#endif //PVA_ALGOS_OBJECT_DETECTOR