Opencv cuda fix

This commit is contained in:
hchandirasekar
2021-03-31 03:44:51 -07:00
parent 516017cc8f
commit 29b0c551a5
2 changed files with 2 additions and 4 deletions
+1 -1
View File
@@ -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';
+1 -3
View File
@@ -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>