|
NVIDIA DeepStream SDK API Reference
|
6.4 Release
|
Go to the documentation of this file.
65 class Yolo :
public IModelParser {
77 nvinfer1::IBuilder* builder, nvinfer1::IBuilderConfig* config);
97 std::vector<float>& weights, nvinfer1::INetworkDefinition& network);
98 std::vector<std::map<std::string, std::string>> parseConfigFile(
99 const std::string cfgFilePath);
100 void parseConfigBlocks();
101 void destroyNetworkUtils();
const std::string m_InputBlobName
std::string configFilePath
NvDsInferStatus parseModel(nvinfer1::INetworkDefinition &network) override
Yolo(const NetworkInfo &networkInfo)
Holds information about an output tensor of the yolo network.
std::vector< float > anchors
std::vector< nvinfer1::Weights > m_TrtWeights
bool hasFullDimsSupported() const override
const std::string m_ConfigFilePath
std::vector< uint > masks
const std::string m_DeviceType
std::string inputBlobName
const char * getModelName() const override
nvinfer1::ICudaEngine * createEngine(nvinfer1::IBuilder *builder, nvinfer1::IBuilderConfig *config)
std::vector< TensorInfo > m_OutputTensors
Holds all the file paths required to build a network.
std::vector< std::map< std::string, std::string > > m_ConfigBlocks
const std::string m_NetworkType
const std::string m_WtsFilePath
NvDsInferStatus
Enum for the status codes returned by NvDsInferContext.