因其工作需要使用C++读取rtsp流,本文将针对linux下c++版读取视频流保存视频,凌乱且各种错误。现将将其成功执行的流程和代码记录,顺带也写了一份python读取rtsp方法,供自己使用及有需之人参考。本文分三部分,第一部分呈现CMakeLists.txt与rtsp_video源码;第二部分插图给出使用方法;第三部分顺带给出python方法的源码。
一.CMakeLists.txt与rtsp_video源码
CMakeLists.txt
cmake_minimum_required(VERSION 2.6)project(rtsp)find_package( OpenCV REQUIRED )include_directories( ${OpenCV_INCLUDE_DIRS} ) add_executable(rtsp_demo rtsp_video.cpp)target_link_libraries( rtsp_demo ${OpenCV_LIBS} )
rtsp_video.cpp
#include #include #include #include using namespace cv;using namespace std;int main() { std::string rtspPath = "rtsp://admin:Vrc123456@192.168.2.226:554"; VideoCapture cap; cap.open(rtspPath); if (!cap.isOpened()) { cout << "cannot open video!" << endl; return 0; } VideoWriter writer = VideoWriter("./rtsp-video.avi",//path and filename (int)cap.get(CAP_PROP_FOURCC), (int)cap.get(CAP_PROP_FPS), Size((int)cap.get(CAP_PROP_FRAME_WIDTH), (int)cap.get(CAP_PROP_FRAME_HEIGHT)), true//colorfull pic ); if (!writer.isOpened()) { cout << "create vedio failed!" <> frame; if (frame.empty()) { cout << "frame is empty!" << endl; break; } cv::Mat img = frame; //cv::imshow("frame", img); writer<<img; //waitKey(1); } return 0;}
注:opencv安装参考网络方法
二.使用方法
步骤1:
打开rtsp_video.cpp文件修改流接口:
std::string rtspPath = “rtsp://admin:Vrc123456@192.168.2.226:554”;
步骤2:
将整个文件夹传入需编译服务器,文件内容包括红框内容:
以下类似编译过程
步骤3:执行
cmake .
make
可出现rtsp_demo,以下红框
无需录视频,执行 ctrl+c 按键
保存视频为rtsp-video.avi
三.python方法的源码
import cv2cap = cv2.VideoCapture('rtsp://admin:Vrc123456@192.168.2.226:554')fourcc = cv2.VideoWriter_fourcc(*'XVID')size = (int(cap.get(cv2.CAP_PROP_FRAME_WIDTH)), int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT)))out = cv2.VideoWriter('./rtsp_video.avi', fourcc,10.0, size)while True: ret, frame = cap.read() out.write(frame) cv2.imshow('frame', frame) cv2.waitKey(1)cap.release()out.release()cv2.destroyAllWindows()
处理算法通用的辅助的code,如读取txt文件,读取xml文件,将xml文件转换成txt文件,读取json文件等