Add GPU version for yolo3detection

Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>

include/tkDNN/DetectionNN.h

@@ -3,7 +3,7 @@
 
 #include <iostream>
 #include <signal.h>
-#include <stdlib.h>     /* srand, rand */
+#include <stdlib.h>    
 #include <unistd.h>
 #include <mutex>
 #include "utils.h"
@@ -14,7 +14,7 @@
 
 #include "tkdnn.h"
 
-// #define OPENCV_CUDA //if OPENCV has been compiled with CUDA and contrib.
+#define OPENCV_CUDA //if OPENCV has been compiled with CUDA and contrib.
 
 namespace tk { namespace dnn {
 

include/tkDNN/Yolo3Detection.h

@@ -17,6 +17,8 @@ private:
     tk::dnn::Yolo* yolo[3];
 
     tk::dnn::Yolo* getYoloLayer(int n=0);
+
+    cv::Mat bgr_h;
     
 public:
     Yolo3Detection() {};