公司网站制作的教程,郑州网站优化公司排名,浙江建设部网站,昌吉市建设局网站前言
在使用yolov5做物体检测中#xff0c;需要拉取视频流。分解任务第一步则是需要使用opencv读取rtsp流#xff0c;只要拿到每一帧图片在进行推理显示即可。
代码
import cv2
def read_rtsp():cap cv2.VideoCapture(rtsp://admin:Vrc123456192.168.2.226:554)fourcc c…前言
在使用yolov5做物体检测中需要拉取视频流。分解任务第一步则是需要使用opencv读取rtsp流只要拿到每一帧图片在进行推理显示即可。
代码
import cv2
def read_rtsp():cap cv2.VideoCapture(rtsp://admin:Vrc123456192.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()
if __name__ __main__:read_rtsp()
关键接口总结
//读取流
cap cv2.VideoCapture(rtsp://...)
//读取每一帧
ret, frame cap.read()
