Alexey
2017-11-03 a4e1dc907875d1a9aaf5800ab33558e9ff0ff649
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
@@ -17,11 +19,13 @@
#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
@@ -70,7 +74,7 @@
   std::ifstream file(filename);
   std::vector<std::string> file_lines;
   if (!file.is_open()) return file_lines;
   for(std::string line; file >> line;) file_lines.push_back(line);
   for(std::string line; getline(file, line);) file_lines.push_back(line);
   std::cout << "object names loaded \n";
   return file_lines;
}
@@ -81,7 +85,7 @@
   std::string filename;
   if (argc > 1) filename = argv[1];
   Detector detector("yolo-voc.cfg", "yolo-voc.weights");
   Detector detector("cfg/yolo-voc.cfg", "yolo-voc.weights");
   auto obj_names = objects_names_from_file("data/voc.names");
   std::string out_videofile = "result.avi";
@@ -98,7 +102,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 +220,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(); }