| | |
| | | #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 |
| | | |
| | |
| | | #include "opencv2/core/version.hpp" |
| | | #ifndef CV_VERSION_EPOCH |
| | | #include "opencv2/videoio/videoio.hpp" |
| | | #pragma comment(lib, "opencv_world320.lib") |
| | | #define OPENCV_VERSION CVAUX_STR(CV_VERSION_MAJOR)""CVAUX_STR(CV_VERSION_MINOR)""CVAUX_STR(CV_VERSION_REVISION) |
| | | #pragma comment(lib, "opencv_world" OPENCV_VERSION ".lib") |
| | | #else |
| | | #pragma comment(lib, "opencv_core2413.lib") |
| | | #pragma comment(lib, "opencv_imgproc2413.lib") |
| | | #pragma comment(lib, "opencv_highgui2413.lib") |
| | | #define OPENCV_VERSION CVAUX_STR(CV_VERSION_EPOCH)""CVAUX_STR(CV_VERSION_MAJOR)""CVAUX_STR(CV_VERSION_MINOR) |
| | | #pragma comment(lib, "opencv_core" OPENCV_VERSION ".lib") |
| | | #pragma comment(lib, "opencv_imgproc" OPENCV_VERSION ".lib") |
| | | #pragma comment(lib, "opencv_highgui" OPENCV_VERSION ".lib") |
| | | #endif |
| | | |
| | | |
| | |
| | | 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; |
| | |
| | | auto img = detector.load_image(filename); |
| | | std::vector<bbox_t> result_vec = detector.detect(img); |
| | | detector.free_image(img); |
| | | show_result(result_vec, obj_names); |
| | | show_console_result(result_vec, obj_names); |
| | | #endif |
| | | } |
| | | catch (std::exception &e) { std::cerr << "exception: " << e.what() << "\n"; getchar(); } |