Add opencv script, modified macro, add opencv section into readme

Signed-off-by: Micaela Verucchi <micaela.verucchi@unimore.it>
2020-04-06 17:25:53 +02:00
parent dfdeb8b36e
commit 674c61d280
6 changed files with 17 additions and 9 deletions
@@ -17,7 +17,7 @@

 #include "tkdnn.h"

-#define OPENCV_CUDA //if OPENCV has been compiled with CUDA and contrib.
+#define OPENCV_CUDACONTRIB //if OPENCV has been compiled with CUDA and contrib.

 namespace tk { namespace dnn {

@@ -37,7 +37,7 @@ class DetectionNN {

        cv::Scalar colors[256];

-#ifdef OPENCV_CUDA
+#ifdef OPENCV_CUDACONTRIB
        cv::cuda::GpuMat bgr[3];
        cv::cuda::GpuMat imagePreproc;
 #else