Fix compile errors dut to Int8BatchStream

Signed-off-by: Micaela Verucchi <micaela.verucchi@unimore.it> Davide Sapienza <sapienza.dav@gmail.com>
2020-04-06 19:11:39 +02:00
parent 8575059666
commit 455291b6dd
3 changed files with 18 additions and 12 deletions
@@ -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 {