infer_post_datatypes.h

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: Copyright (c) 2020 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 * SPDX-License-Identifier: LicenseRef-NvidiaProprietary
 *
 * NVIDIA CORPORATION, its affiliates and licensors retain all intellectual
 * property and proprietary rights in and to this material, related
 * documentation and any modifications thereto. Any use, reproduction,
 * disclosure or distribution of this material and related documentation
 * without an express license agreement from NVIDIA CORPORATION or
 * its affiliates is strictly prohibited.
 */
 
#ifndef __INFER_POST_DATATYPES_H__
#define __INFER_POST_DATATYPES_H__
 
#include <string>
#include <nvdsinfer.h>
 
typedef struct {
    float left;
    float top;
    float width;
    float height;
    /* Index for the object class. */
    int classIndex;
    /* String label for the detected object. */
    char* label;
    /* confidence score of the detected object. */
    float confidence;
} NvDsInferObject;
 
typedef struct {
    NvDsInferObject* objects;
    unsigned int numObjects;
} NvDsInferDetectionOutput;
 
#ifdef __cplusplus
 
struct InferAttribute : NvDsInferAttribute {
    /* NvDsInferAttribute::attributeLabel would be ignored */
    std::string safeAttributeLabel;
};
 
typedef struct {
    std::vector<InferAttribute> attributes;
    std::string label;
} InferClassificationOutput;
 
#endif
 
typedef struct {
    unsigned int width;
    unsigned int height;
    unsigned int classes;
    int* class_map;
    float* class_probability_map;
} NvDsInferSegmentationOutput;
 
struct TritonClassParams {
    uint32_t topK = 0;
    float threshold = 0.0f;
    std::string tensorName;
};
 
#define INFER_SERVER_PRIVATE_BUF "@@NvInferServer"
 
#define INFER_SERVER_DETECTION_BUF_NAME INFER_SERVER_PRIVATE_BUF "Detections"
#define INFER_SERVER_CLASSIFICATION_BUF_NAME \
    INFER_SERVER_PRIVATE_BUF "Classfications"
#define INFER_SERVER_SEGMENTATION_BUF_NAME \
    INFER_SERVER_PRIVATE_BUF "Segmentations"
 
#endif