Fix INT8 calibration 2

2021-11-26 17:38:30 -03:00
parent b6a7ec5d15
commit dcc44b730c
1 changed files with 5 additions and 0 deletions
--- a/nvdsinfer_custom_impl_Yolo/calibrator.cpp
+++ b/nvdsinfer_custom_impl_Yolo/calibrator.cpp
@@ -30,6 +30,11 @@ namespace nvinfer1
            delete[] batchData;
    }

+    int Int8EntropyCalibrator2::getBatchSize() const noexcept
+    {
+        return batchSize;
+    }
+
    bool Int8EntropyCalibrator2::getBatch(void **bindings, const char **names, int nbBindings) noexcept
    {
        if (imageIndex + batchSize > uint(imgPaths.size()))