python获取RTSP视频流的截图

因为公司有个一个项目需要获取的到当前视频的当前帧,用php没找到很好的解决方法,遂用python的方法获取到截图,以下代码

import uuid
import cv2
import sys

# RTSP="rtsp://admin:123qwe123@192.168.110.64:554"
CAMID="kk"
# 存视频图像
RTSP = sys.argv[1] #获取传过来的第一个参数
ROAD = sys.argv[2] #获取传过来的第一个参数
VideoCap = cv2.VideoCapture(RTSP)
ret, frame = VideoCap.read()
if ret:
   # FrameName = "_" + str(CAMID) + "_" + str(uuid.uuid1())
   FrameName = str(uuid.uuid1())
   # cv2.imwrite('D:/Pecker/imgs/%s.jpg' % FrameName, frame)
   cv2.imwrite(ROAD+'%s.png' % FrameName, frame)
   # print("成功截图", FrameName)
   print(FrameName+'.png')
VideoCap.release()

cv2需要安装

pip install opencv-python -i https://pypi.tuna.tsinghua.edu.cn/simple