9.1/sources/includes/nvdsinferserver/infer_post_datatypes.h

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: Copyright (c) 2020-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 * SPDX-License-Identifier: Apache-2.0
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
 
#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