From 482e1226552c752ab2b56d3510c8d2fd1ad0ca13 Mon Sep 17 00:00:00 2001
From: Davide Sapienza <sapienza.dav@gmail.com>
Date: Fri, 27 Mar 2020 10:43:49 +0100
Subject: [PATCH] Fix opencv cuda include error

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
---
 include/tkDNN/DetectionNN.h | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/include/tkDNN/DetectionNN.h b/include/tkDNN/DetectionNN.h
index 2005d9b..e09418a 100644
--- a/include/tkDNN/DetectionNN.h
+++ b/include/tkDNN/DetectionNN.h
@@ -11,6 +11,9 @@
 #include <opencv2/core/core.hpp>
 #include <opencv2/highgui/highgui.hpp>
 #include <opencv2/imgproc/imgproc.hpp>
+#include <opencv2/cudawarping.hpp>
+#include <opencv2/cudaarithm.hpp>
+
 
 #include "tkdnn.h"