New features

- Added support for INT8 calibration
- Added support for non square models
- Updated mAP comparison between models
This commit is contained in:
Marcos Luciano
2021-06-18 00:30:10 -03:00
parent 312e9a448d
commit cbd9675dc2
74 changed files with 3287 additions and 700 deletions

View File

@@ -40,6 +40,8 @@ struct NetworkInfo
std::string networkType;
std::string configFilePath;
std::string wtsFilePath;
std::string int8CalibPath;
std::string networkMode;
std::string deviceType;
std::string inputBlobName;
};
@@ -48,7 +50,8 @@ struct TensorInfo
{
std::string blobName;
uint stride{0};
uint gridSize{0};
uint gridSizeY{0};
uint gridSizeX{0};
uint numClasses{0};
uint numBBoxes{0};
uint64_t volume{0};
@@ -75,6 +78,8 @@ protected:
const std::string m_NetworkType;
const std::string m_ConfigFilePath;
const std::string m_WtsFilePath;
const std::string m_Int8CalibPath;
const std::string m_NetworkMode;
const std::string m_DeviceType;
const std::string m_InputBlobName;
std::vector<TensorInfo> m_OutputTensors;
@@ -84,6 +89,7 @@ protected:
uint m_InputW;
uint m_InputC;
uint64_t m_InputSize;
uint m_LetterBox;
std::vector<nvinfer1::Weights> m_TrtWeights;