AlexeyAB
2017-10-18 437fbb4b3a1168a0ce69525f17d0a4b31e864ec7
src/yolo_console_dll.cpp
@@ -8,7 +8,9 @@
#include <mutex>              // std::mutex, std::unique_lock
#include <condition_variable> // std::condition_variable
#ifdef WIN32
#define OPENCV
#endif
#include "yolo_v2_class.hpp"  // imported functions from DLL
@@ -98,7 +100,7 @@
         std::string const file_ext = filename.substr(filename.find_last_of(".") + 1);
         std::string const protocol = filename.substr(0, 7);
         if (file_ext == "avi" || file_ext == "mp4" || file_ext == "mjpg" || file_ext == "mov" ||  // video file
            protocol == "rtsp://" || protocol == "http://" || protocol == "https:/")   // video network stream
            protocol == "rtmp://" || protocol == "rtsp://" || protocol == "http://" || protocol == "https:/")  // video network stream
         {
            cv::Mat cap_frame, cur_frame, det_frame, write_frame;
            std::shared_ptr<image_t> det_image;