NvInferPluginUtils.h

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: Copyright (c) 1993-2024 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 NV_INFER_PLUGIN_UTILS_H
#define NV_INFER_PLUGIN_UTILS_H
 
#include "NvInferRuntimeCommon.h"
 
 
namespace nvinfer1
{
namespace plugin
{
 
struct PriorBoxParameters
{
    float *minSize;          
    float *maxSize;          
    float *aspectRatios;     
    int32_t numMinSize;      
    int32_t numMaxSize;      
    int32_t numAspectRatios; 
    bool flip;               
    bool clip;               
    float variance[4];       
    int32_t imgH;            
    int32_t imgW;            
    float stepH;             
    float stepW;             
    float offset;            
};
 
struct RPROIParams
{
    int32_t poolingH;          
    int32_t poolingW;          
    int32_t featureStride;     
    int32_t preNmsTop;         
    int32_t nmsMaxOut;         
    int32_t anchorsRatioCount; 
    int32_t anchorsScaleCount; 
    float iouThreshold;        
    float minBoxSize;          
    float spatialScale;        
};
 
struct GridAnchorParameters
{
    float minSize;           
    float maxSize;           
    float* aspectRatios;     
    int32_t numAspectRatios; 
    int32_t H;               
    int32_t W;               
    float variance[4];       
};
 
enum class CodeTypeSSD : int32_t
{
    CORNER TRT_DEPRECATED_ENUM = 0,      
    CENTER_SIZE TRT_DEPRECATED_ENUM = 1, 
    CORNER_SIZE TRT_DEPRECATED_ENUM = 2, 
    TF_CENTER TRT_DEPRECATED_ENUM = 3    
};
 
struct TRT_DEPRECATED DetectionOutputParameters
{
    bool shareLocation;           
    bool varianceEncodedInTarget; 
    int32_t backgroundLabelId;    
    int32_t numClasses;           
    int32_t topK;                 
    int32_t keepTopK;             
    float confidenceThreshold;    
    float nmsThreshold;           
    CodeTypeSSD codeType;         
    int32_t inputOrder[3];        
    bool confSigmoid;             
    bool isNormalized;            
    bool isBatchAgnostic{true};   
};
 
struct softmaxTree
{
    int32_t* leaf;
    int32_t n;
    int32_t* parent;
    int32_t* child;
    int32_t* group;
    char** name;
    int32_t groups;
    int32_t* groupSize;
    int32_t* groupOffset;
};
 
struct RegionParameters
{
    int32_t num;         
    int32_t coords;      
    int32_t classes;     
    softmaxTree* smTree; 
};
 
struct TRT_DEPRECATED NMSParameters
{
    bool shareLocation;        
    int32_t backgroundLabelId; 
    int32_t numClasses;        
    int32_t topK;              
    int32_t keepTopK;          
    float scoreThreshold;      
    float iouThreshold;        
    bool isNormalized;         
};
 
} // namespace plugin
} // namespace nvinfer1
 
#endif // NV_INFER_PLUGIN_UTILS_H