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"