diff --git a/src/cv.cpp b/src/cv.cpp index 5fdb035..b812d71 100644 --- a/src/cv.cpp +++ b/src/cv.cpp @@ -26,6 +26,8 @@ void initCV() { facemark->loadModel (resolvePath("cvdata/lbfmodel.yaml")); vid = cv::VideoCapture (0); + + if(!vid.isOpened()) {std::cout << "Camera not opened!" << std::endl;} } void dnnFaceDetect(cv::Mat inFrame, std::vector* faces) { @@ -56,6 +58,8 @@ void dnnFaceDetect(cv::Mat inFrame, std::vector* faces) { void cvFrame() { vid.read(frame); + if(frame.empty()) return; + cv::cvtColor (frame, gray, cv::COLOR_BGR2GRAY); std::vector faces; @@ -157,6 +161,8 @@ void cvFrame() { } void cvShowFrame() { + if(frame.empty()) return; + cv::imshow("Video Input", frame); cv::waitKey(1); }