Add GPU version for yolo3detection

Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-03-23 21:29:29 +01:00
parent df37e11709
commit 3eda9b9219
3 changed files with 11 additions and 5 deletions
@@ -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 {