| | |
| | | #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 |
| | | |
| | |
| | | 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; |