nvdsinfer.h

Go to the documentation of this file.

/*
 * Copyright (c) 2017-2020, 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 _NVDSINFER_H_
#define _NVDSINFER_H_
 
#include <stdint.h>
 
#ifdef __cplusplus
extern "C"
{
#endif
 
#define NVDSINFER_MAX_DIMS 8
 
#define _DS_DEPRECATED_(STR) __attribute__ ((deprecated (STR)))
 
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;
  union {
      NvDsInferDims inferDims;
      NvDsInferDims dims _DS_DEPRECATED_("dims is deprecated. Use inferDims instead");
  };
  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)
 
#define getDimsHWCFromDims(dimsCHW,dims) \
  do { \
    (dimsCHW).h = (dims).d[0]; \
    (dimsCHW).w = (dims).d[1]; \
    (dimsCHW).c = (dims).d[2]; \
  } while (0)
 
typedef struct
{
  unsigned int classId;
 
  float left;
  float top;
  float width;
  float height;
 
  float detectionConfidence;
} NvDsInferObjectDetectionInfo;
 
typedef NvDsInferObjectDetectionInfo NvDsInferParseObjectInfo;
 
typedef struct
{
  unsigned int classId;
 
  float left;
  float top;
  float width;
  float height;
 
  float detectionConfidence;
 
  float *mask;
  unsigned int mask_width;
  unsigned int mask_height;
  unsigned int mask_size;
} NvDsInferInstanceMaskInfo;
 
typedef struct
{
    unsigned int attributeIndex;
    unsigned int attributeValue;
    float attributeConfidence;
    char *attributeLabel;
} NvDsInferAttribute;
 
typedef enum {
    NVDSINFER_SUCCESS = 0,
    NVDSINFER_CONFIG_FAILED,
    NVDSINFER_CUSTOM_LIB_FAILED,
    NVDSINFER_INVALID_PARAMS,
    NVDSINFER_OUTPUT_PARSING_FAILED,
    NVDSINFER_CUDA_ERROR,
    NVDSINFER_TENSORRT_ERROR,
    NVDSINFER_RESOURCE_ERROR,
    NVDSINFER_TRITON_ERROR,
    NVDSINFER_TRTIS_ERROR = NVDSINFER_TRITON_ERROR,
    NVDSINFER_UNKNOWN_ERROR
} NvDsInferStatus;
 
typedef enum {
    NVDSINFER_LOG_ERROR = 0,
    NVDSINFER_LOG_WARNING,
    NVDSINFER_LOG_INFO,
    NVDSINFER_LOG_DEBUG,
} NvDsInferLogLevel;
 
const char* NvDsInferStatus2Str(NvDsInferStatus status);
 
#ifdef __cplusplus
}
#endif
 
/* C++ data types */
#ifdef __cplusplus
 
typedef enum
{
    kSELECTOR_MIN = 0,
    kSELECTOR_OPT,
    kSELECTOR_MAX,
    kSELECTOR_SIZE
} NvDsInferProfileSelector;
 
typedef struct
{
    int batchSize = 0;
    NvDsInferDims dims = {0};
} NvDsInferBatchDims;
 
struct NvDsInferBatchDimsLayerInfo : NvDsInferLayerInfo
{
    NvDsInferBatchDims profileDims[kSELECTOR_SIZE];
};
 
#endif
 
#endif