New features
- Added support for INT8 calibration - Added support for non square models - Updated mAP comparison between models
This commit is contained in:
@@ -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;
|
||||
|
||||
|
||||
Reference in New Issue
Block a user