Merge branch 'master' of https://github.com/eric612/MobileNet-YOLO

DancingJane · Aug 21, 2019 · 96bb265 · 96bb265
2 parents de10a33 + 53e1588
commit 96bb265
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/caffe/util/im_transforms.cpp b/src/caffe/util/im_transforms.cpp
@@ -1,7 +1,7 @@
 #ifdef USE_OPENCV
 #include <opencv2/highgui/highgui.hpp>
 
-#if CV_VERSION_MAJOR == 3
+#if CV_VERSION_MAJOR >= 3
 #include <opencv2/imgcodecs/imgcodecs.hpp>
 #define CV_GRAY2BGR cv::COLOR_GRAY2BGR
 #define CV_BGR2GRAY cv::COLOR_BGR2GRAY