From 2e92944f1da276ec31eeb08bd99cf7e78679b23c Mon Sep 17 00:00:00 2001
From: hchandirasekar <f20180523@goa.bits-pilani.ac.in>
Date: Wed, 31 Mar 2021 03:44:51 -0700
Subject: [PATCH] Opencv cuda fix

---
 demo/demo/demo.cpp          | 2 +-
 include/tkDNN/DetectionNN.h | 4 +---
 2 files changed, 2 insertions(+), 4 deletions(-)

diff --git a/demo/demo/demo.cpp b/demo/demo/demo.cpp
index f97ead9..59bb1a2 100644
--- a/demo/demo/demo.cpp
+++ b/demo/demo/demo.cpp
@@ -25,7 +25,7 @@ int main(int argc, char *argv[]) {
     std::string net = "yolo4tiny_fp32.rt";
     if(argc > 1)
         net = argv[1]; 
-    std::string input = "../demo/yolo_test.mp4";
+    std::string input = "..\..\..\demo\yolo_test.mp4";
     if(argc > 2)
         input = argv[2]; 
     char ntype = 'y';
diff --git a/include/tkDNN/DetectionNN.h b/include/tkDNN/DetectionNN.h
index fb29be1..b1266e0 100644
--- a/include/tkDNN/DetectionNN.h
+++ b/include/tkDNN/DetectionNN.h
@@ -6,8 +6,6 @@
 #include <stdlib.h>    
 #ifdef __linux__
 #include <unistd.h>
-#elif _WIN32
-#include <Windows.h>
 #endif 
 
 #include <mutex>
@@ -19,7 +17,7 @@
 
 #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>