Fixed yolo_console_dll.cpp for RTMP protocol

pull/234/head
AlexeyAB 8 years ago
parent 25750d59cc
commit 5a2efd5e53
  1. 2
      src/yolo_console_dll.cpp

@ -98,7 +98,7 @@ int main(int argc, char *argv[])
std::string const file_ext = filename.substr(filename.find_last_of(".") + 1); std::string const file_ext = filename.substr(filename.find_last_of(".") + 1);
std::string const protocol = filename.substr(0, 7); std::string const protocol = filename.substr(0, 7);
if (file_ext == "avi" || file_ext == "mp4" || file_ext == "mjpg" || file_ext == "mov" || // video file 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; cv::Mat cap_frame, cur_frame, det_frame, write_frame;
std::shared_ptr<image_t> det_image; std::shared_ptr<image_t> det_image;

Loading…
Cancel
Save