mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity

Opencv cuda fix

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files
demo/demo/demo.cpp
+1 -1
View File
@@ -25,7 +25,7 @@ int main(int argc, char *argv[]) {
std::string net = "yolo4tiny_fp32.rt"; std::string net = "yolo4tiny_fp32.rt";
if(argc > 1) if(argc > 1)
net = argv[1]; net = argv[1];
std::string input = "../demo/yolo_test.mp4"; std::string input = "..\..\..\demo\yolo_test.mp4";
if(argc > 2) if(argc > 2)
input = argv[2]; input = argv[2];
char ntype = 'y'; char ntype = 'y';
include/tkDNN/DetectionNN.h
+1 -3
View File
@@ -6,8 +6,6 @@
#include <stdlib.h> #include <stdlib.h>
#ifdef __linux__ #ifdef __linux__
#include <unistd.h> #include <unistd.h>
#elif _WIN32
#include <Windows.h>
#endif #endif
#include <mutex> #include <mutex>
@@ -19,7 +17,7 @@
#include "tkdnn.h" #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 #ifdef OPENCV_CUDACONTRIB
#include <opencv2/cudawarping.hpp> #include <opencv2/cudawarping.hpp>