New features

- Added support for INT8 calibration
- Added support for non square models
- Updated mAP comparison between models

native/nvdsinfer_custom_impl_Yolo/calibrator.h

@@ -0,0 +1,62 @@
+/*
+ * Created by Marcos Luciano
+ * https://www.github.com/marcoslucianops
+ */
+
+#ifndef CALIBRATOR_H
+#define CALIBRATOR_H
+
+#include "opencv2/opencv.hpp"
+#include "cuda_runtime.h"
+#include "NvInfer.h"
+#include <vector>
+#include <string>
+
+#ifndef CUDA_CHECK
+#define CUDA_CHECK(callstr)                                                                    \
+    {                                                                                          \
+        cudaError_t error_code = callstr;                                                      \
+        if (error_code != cudaSuccess) {                                                       \
+            std::cerr << "CUDA error " << error_code << " at " << __FILE__ << ":" << __LINE__; \
+            assert(0);                                                                         \
+        }                                                                                      \
+    }
+#endif
+
+namespace nvinfer1 {
+    class int8EntroyCalibrator : public nvinfer1::IInt8EntropyCalibrator2 {
+    public:
+        int8EntroyCalibrator(const int &batchsize,
+                             const int &channels,
+                             const int &height,
+                             const int &width,
+                             const int &letterbox,
+                             const std::string &imgPath,
+                             const std::string &calibTablePath);
+
+        virtual ~int8EntroyCalibrator();
+        int getBatchSize() const override { return batchSize; }
+        bool getBatch(void *bindings[], const char *names[], int nbBindings) override;
+        const void *readCalibrationCache(std::size_t &length) override;
+        void writeCalibrationCache(const void *ptr, std::size_t length) override;
+
+    private:
+        int batchSize;
+        int inputC;
+        int inputH;
+        int inputW;
+        int letterBox;
+        std::string calibTablePath;
+        size_t imageIndex;
+        size_t inputCount;
+        std::vector<std::string> imgPaths;
+        float *batchData{ nullptr };
+        void  *deviceInput{ nullptr };
+        bool readCache;
+        std::vector<char> calibrationCache;
+    };
+}
+
+std::vector<float> prepareImage(cv::Mat& img, int input_c, int input_h, int input_w, int letter_box);
+
+#endif //CALIBRATOR_H