Fix the INT8 calibrator sintax

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>

include/tkDNN/Int8BatchStream.h

@@ -19,16 +19,21 @@
 #include "utils.h"
 #include "tkdnn.h"
 
-class BatchStream
-{
+/*
+ * BatchStream implements the stream for the INT8 calibrator. 
+ * It reads the two files .txt with the list of image file names 
+ * and the list of label file names. 
+ * It then iterates on images and labels.
+ */
+class BatchStream {
 public:
 	BatchStream(tk::dnn::dataDim_t dim, int batchSize, int maxBatches, const std::string& fileimglist, const std::string& filelabellist);
-	virtual ~BatchStream() {}
+	virtual ~BatchStream() { }
 	void reset(int firstBatch);
 	bool next();
 	void skip(int skipCount);
-	float *getBatch() { return mBatch.data();}
-	float *getLabels() { return mLabels.data();}
+	float *getBatch() { return mBatch.data(); }
+	float *getLabels() { return mLabels.data(); }
 	int getBatchesRead() const { return mBatchCount; }
 	int getBatchSize() const { return mBatchSize; }
 	nvinfer1::DimsNCHW getDims() const { return mDims; }
@@ -43,7 +48,8 @@ private:
 	int mBatchSize{ 0 };
 	int mMaxBatches{ 0 };
 	int mBatchCount{ 0 };
-	int mFileCount{ 0 }, mFileBatchPos{ 0 };
+	int mFileCount{ 0 };
+	int mFileBatchPos{ 0 };
 	int mImageSize{ 0 };
 
 	nvinfer1::DimsNCHW mDims;

include/tkDNN/Int8Calibrator.h

@@ -18,9 +18,17 @@
 #include "tkdnn.h"
 #include "utils.h"
 
-class Int8EntropyCalibrator : public nvinfer1::IInt8EntropyCalibrator{
+/*
+ * Int8EntropyCalibrator implements the INT8 calibrator to achieve the
+ * INT8 quantization. It uses a BatchStream stream to scroll through 
+ * images data. It also implements the calibration cache, a way to 
+ * save the calibration process results to reduce the running time: 
+ * the calibration process takes a long time.
+ */
+class Int8EntropyCalibrator : public nvinfer1::IInt8EntropyCalibrator {
 public:
-	Int8EntropyCalibrator(BatchStream& stream, int firstBatch, const std::string& calibTableFilePath, const std::string& inputBlobName, bool readCache = true);
+	Int8EntropyCalibrator(BatchStream& stream, int firstBatch, const std::string& calibTableFilePath, 
+							const std::string& inputBlobName, bool readCache = true);
 	virtual ~Int8EntropyCalibrator() { checkCuda(cudaFree(mDeviceInput)); }
 	int getBatchSize() const override { return mStream.getBatchSize(); }
 	bool getBatch(void* bindings[], const char* names[], int nbBindings) override;
@@ -29,7 +37,7 @@ public:
 
 private:
 	BatchStream mStream;
-	const std::string mCalibTableFilePath{nullptr};
+	const std::string mCalibTableFilePath{ nullptr };
 	const std::string mInputBlobName;
 	bool mReadCache{ true };