diff --git a/lib/src/detector.cc b/lib/src/detector.cc index 2114a9a..00ce1e6 100644 --- a/lib/src/detector.cc +++ b/lib/src/detector.cc @@ -14,9 +14,17 @@ std::vector MTCNNDetector::detect(const cv::Mat &img, cv::Mat rgbImg; if (img.channels() == 3) { - cv::cvtColor(img, rgbImg, CV_BGR2RGB); + if (CV_VERSION_MAJOR >= 4) { + cv::cvtColor(img, rgbImg, cv::COLOR_BGR2RGB); + } else { + cv::cvtColor(img, rgbImg, CV_BGR2RGB); + } } else if (img.channels() == 4) { - cv::cvtColor(img, rgbImg, CV_BGRA2RGB); + if (CV_VERSION_MAJOR >= 4) { + cv::cvtColor(img, rgbImg, cv::COLOR_BGRA2RGB); + } else { + cv::cvtColor(img, rgbImg, CV_BGRA2RGB); + } } if (rgbImg.empty()) { return std::vector(); diff --git a/sample/CMakeLists.txt b/sample/CMakeLists.txt index 36aafe3..0a7e344 100644 --- a/sample/CMakeLists.txt +++ b/sample/CMakeLists.txt @@ -9,7 +9,7 @@ set (SAMPLE_INCLUDE_PATH "${SAMPLE_PATH}/include") find_package(OpenCV REQUIRED) set(Boost_USE_STATIC_LIBS ON) -FIND_PACKAGE(Boost 1.58 COMPONENTS filesystem timer REQUIRED) +FIND_PACKAGE(Boost 1.58 COMPONENTS filesystem timer REQUIRED chrono) #set includes include_directories (