Fix compile errors dut to Int8BatchStream

Signed-off-by: Micaela Verucchi <micaela.verucchi@unimore.it>
               Davide Sapienza <sapienza.dav@gmail.com>

include/tkDNN/DetectionNN.h

@@ -11,13 +11,16 @@
 #include <opencv2/core/core.hpp>
 #include <opencv2/highgui/highgui.hpp>
 #include <opencv2/imgproc/imgproc.hpp>
-#include <opencv2/cudawarping.hpp>
-#include <opencv2/cudaarithm.hpp>
-
 
 #include "tkdnn.h"
 
-#define OPENCV_CUDACONTRIB //if OPENCV has been compiled with CUDA and contrib.
+//#define OPENCV_CUDACONTRIB //if OPENCV has been compiled with CUDA and contrib.
+
+#ifdef OPENCV_CUDACONTRIB
+#include <opencv2/cudawarping.hpp>
+#include <opencv2/cudaarithm.hpp>
+#endif
+
 
 namespace tk { namespace dnn {
 

include/tkDNN/Int8BatchStream.h

@@ -8,18 +8,16 @@
 #include <stdint.h>
 #include <iostream>
 #include <string>
-#include "NvInfer.h"
-
 #include <fstream>
 #include <iomanip>
+#include <signal.h>
+#include <stdlib.h>    
+#include <unistd.h>
+#include <mutex>
 
-#include <opencv2/core/core.hpp>
+#include "NvInfer.h"
-#include <opencv2/dnn/dnn.hpp>
-#include <opencv2/highgui/highgui.hpp>
-#include <opencv2/imgproc/imgproc.hpp>
-
-#include "tkdnn.h"
 #include "utils.h"
+#include "tkdnn.h"
 
 class BatchStream
 {

src/Int8BatchStream.cpp

@@ -1,5 +1,10 @@
 #include "Int8BatchStream.h"
 
+#include <opencv2/core/core.hpp>
+#include <opencv2/dnn/dnn.hpp>
+#include <opencv2/highgui/highgui.hpp>
+#include <opencv2/imgproc/imgproc.hpp>
+
 BatchStream::BatchStream(tk::dnn::dataDim_t dim, int batchSize, int maxBatches, const std::string& fileimglist, const std::string& filelabellist)
 {
     mBatchSize = batchSize;