thebears/deepstream_yolo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix INT8 calibration 2

Browse Source
This commit is contained in:
unknown
2021-11-26 17:38:30 -03:00
parent b6a7ec5d15
commit dcc44b730c
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
nvdsinfer_custom_impl_Yolo/calibrator.cpp
View File

@@ -30,6 +30,11 @@ namespace nvinfer1
delete[] batchData; delete[] batchData;
} }
int Int8EntropyCalibrator2::getBatchSize() const noexcept
{
return batchSize;
}
bool Int8EntropyCalibrator2::getBatch(void **bindings, const char **names, int nbBindings) noexcept bool Int8EntropyCalibrator2::getBatch(void **bindings, const char **names, int nbBindings) noexcept
{ {
if (imageIndex + batchSize > uint(imgPaths.size())) if (imageIndex + batchSize > uint(imgPaths.size()))