nvdsinfer.h

Go to the documentation of this file.

#ifndef _NVDSINFER_H_
#define _NVDSINFER_H_

#ifdef __cplusplus
extern "C"
{
#endif

#define NVDSINFER_MAX_DIMS 8

typedef struct
{
  unsigned int numDims;
  unsigned int d[NVDSINFER_MAX_DIMS];
  unsigned int numElements;
} NvDsInferDims;

typedef struct
{
  unsigned int c;
  unsigned int h;
  unsigned int w;
} NvDsInferDimsCHW;

typedef enum
{
  FLOAT = 0,
  HALF = 1,
  INT8 = 2,
  INT32 = 3
} NvDsInferDataType;

typedef struct
{
  NvDsInferDataType dataType;
  NvDsInferDims dims;
  int bindingIndex;
  const char* layerName;
  void *buffer;
  int isInput;
} NvDsInferLayerInfo;

typedef struct
{
  unsigned int width;
  unsigned int height;
  unsigned int channels;
} NvDsInferNetworkInfo;

#define getDimsCHWFromDims(dimsCHW,dims) \
  do { \
    (dimsCHW).c = (dims).d[0]; \
    (dimsCHW).h = (dims).d[1]; \
    (dimsCHW).w = (dims).d[2]; \
  } while (0)

typedef struct
{
  unsigned int classId;

  unsigned int left;
  unsigned int top;
  unsigned int width;
  unsigned int height;

  float detectionConfidence;
} NvDsInferObjectDetectionInfo;

typedef NvDsInferObjectDetectionInfo NvDsInferParseObjectInfo;

typedef struct
{
    unsigned int attributeIndex;
    unsigned int attributeValue;
    float attributeConfidence;
    const char *attributeLabel;
} NvDsInferAttribute;

#ifdef __cplusplus
}
#endif

#endif