From dcc44b730ccf39028b54a0b916bbec7bcf1d44ab Mon Sep 17 00:00:00 2001
From: unknown <marcoslucianops@gmail.com>
Date: Fri, 26 Nov 2021 17:38:30 -0300
Subject: [PATCH] Fix INT8 calibration 2

---
 nvdsinfer_custom_impl_Yolo/calibrator.cpp | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/nvdsinfer_custom_impl_Yolo/calibrator.cpp b/nvdsinfer_custom_impl_Yolo/calibrator.cpp
index f5a4408..2cc1e53 100644
--- 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()))