这是为实现ABLR模型结果可视化写的一个demo。环境要求如下:
python3
opencv-python4.1.0
代码如下:
#coding=utf-8
import cv2
print(cv2.__version__)
#打开原完整视频
cap = cv2.VideoCapture(r'E:\深度学习\0605\src\model\s13-d21.avi')#打开视频
if not cap.isOpened():#判断文件是否存在
print ("could not open :",fn)
sys.exit()
length = cap.get(cv2.CAP_PROP_FRAME_COUNT)#获得该视频的帧数
h = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
w = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
fps = int(cap.get(cv2.CAP_PROP_FPS))#获得该视频的帧率
print(fps)
print(length)
size = (w,h)
print(size)
#获取视频片段,保存
videoWriter =cv2.VideoWriter(r'E:\深度学习\0605\src\model\s13-d21_predict.avi',cv2.VideoWriter_fourcc('X','V','I','D'),fps,size)
i = 0
while True:
success,frame = cap.read()
if success:
i += 1
print('i = ',i)
if(i>=1500 and i <= 2500):