python 使用 ffmpeg拉流 储存到本地

rtmp 使用 python 拉流直播和存储都本地

import cv2  # 这里需要依赖的还有 ffmpeg 
#
cap = cv2.VideoCapture('rtmp://58.200.131.2:1935/livetv/hunantv') # 拉流的地址是湖南卫视的
ret, frame = cap.read()
print(ret) # 这里会返回是否正常返回流，正常返回True
fourcc = cv2.VideoWriter_fourcc(*'XVID') # 创建本地文件
fps = cap.get(cv2.CAP_PROP_FPS)
size = (int(cap.get(cv2.CAP_PROP_FRAME_WIDTH)), \
int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT)))

out = cv2.VideoWriter('camera_test.avi', fourcc, fps, size)
while ret:

ret, frame = cap.read()

cv2.imshow("frame", frame)

if cv2.waitKey(1) & 0xFF == ord('q'):

break

# 写入一帧

          out.write(frame)
cv2.destroyAllWindows()
cap.release()