AlexeyAB
2017-10-19 e50489bcd7f02276798ec5c4122a1618d4cfd058
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;
@@ -216,7 +218,7 @@
         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(); }