python+opencv读取rtsp流

前言

在使用yolov5做物体检测中,需要拉取视频流。分解任务第一步则是需要使用opencv读取rtsp流,只要拿到每一帧图片在进行推理显示即可。

代码

复制代码
import cv2
def read_rtsp():
    cap = 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()
if __name__ == '__main__':
    read_rtsp()

关键接口总结

复制代码
//读取流
cap = cv2.VideoCapture('rtsp://...')
//读取每一帧
ret, frame = cap.read()
相关推荐
chao_78921 分钟前
二分查找篇——搜索旋转排序数组【LeetCode】一次二分查找
数据结构·python·算法·leetcode·二分查找
烛阴30 分钟前
Python装饰器解除:如何让被装饰的函数重获自由?
前端·python
noravinsc1 小时前
django 一个表中包括id和parentid,如何通过parentid找到全部父爷id
python·django·sqlite
ajassi20001 小时前
开源 python 应用 开发(三)python语法介绍
linux·python·开源·自动化
沉默媛2 小时前
如何安装python以及jupyter notebook
开发语言·python·jupyter
Deng9452013143 小时前
基于Python的旅游数据可视化应用
python·numpy·pandas·旅游·数据可视化技术
2401_878624793 小时前
pytorch 自动微分
人工智能·pytorch·python·机器学习
胖达不服输3 小时前
「日拱一码」021 机器学习——特征工程
人工智能·python·机器学习·特征工程
screenCui4 小时前
macOS运行python程序遇libiomp5.dylib库冲突错误解决方案
开发语言·python·macos
小眼睛羊羊4 小时前
pyinstaller打包paddleocr
python