NvInferPluginUtils.h

前往此文件的文档。

/*
 * SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES. 保留所有权利。
 * SPDX-License-Identifier: Apache-2.0
 *
 * 根据 Apache License, Version 2.0 ("License") 授权；
 * 除非遵守 License 规定，否则您不得使用此文件。
 * 您可以在以下位置获得 License 副本：
 *
 * https://apache.ac.cn/licenses/LICENSE-2.0
 *
 * 除非适用法律要求或书面同意，否则根据 License 分发的软件
 * 依“按原样”基础分发，
 * 不附带任何形式的明示或暗示的保证或条件。
 * 有关特定语言的权限和限制，请参阅 License。
 * 请参阅 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