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
@@ -43,7 +43,7 @@ private:
    
    float *target_coords;

-    #ifdef OPENCV_CUDA
+    #ifdef OPENCV_CUDACONTRIB
        float *mean_d;
        float *stddev_d;
    #else
@@ -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