OpenCV bug1:生成的视频无法播放
问题描述
用OpenCV截取视频中的某一段,生成新的视频,生成的视频无法播放。
原因分析:
size设置的有问题
源代码
'''
功 能: 截取指定时间段内的视频
'''
import numpy as np
import cv2
import os
import time
START_HOUR = 0
START_MIN = 0
START_SECOND = 1
START_TIME = START_HOUR * 3600 + START_MIN * 60 + START_SECOND # 设置开始时间(单位秒)
END_HOUR = 0
END_MIN = 2
END_SECOND = 8
END_TIME = END_HOUR * 3600 + END_MIN * 60 + END_SECOND # 设置结束时间(单位秒)
video = "C:/dataset/HIT_code_video/video_figure/video/5555.mp4"
cap = cv2.VideoCapture(video)
FPS = cap.get(cv2.CAP_PROP_FPS)
print(FPS)
FPS = 25
size = (1920,1080)
print(size)
TOTAL_FRAME = int(cap.get(cv2.CAP_PROP_FRAME_COUNT)) # 获取视频总帧数
frameToStart = START_TIME * FPS # 开始帧 = 开始时间*帧率
print(frameToStart)
frametoStop = END_TIME * FPS # 结束帧 = 结束时间*帧率
print(frametoStop)
videoWriter =cv2.VideoWriter('C:/dataset/HIT_code_video/video_figure/video/scene1.mp4',cv2.VideoWriter_fourcc('m','p','4','v'),FPS,size)
COUNT = 0
while True:
success, frame = cap.read()
if success:
COUNT += 1
if COUNT <= frametoStop and COUNT > frameToStart: # 选取起始帧
print('correct= ', COUNT)
videoWriter.write(frame)
# print('mistake= ', COUNT)
if COUNT > frametoStop:
break
print('end')
size = (1920,1080) 这里有问题,和原视频的size(1024,576)不相等,这里size必须得相等。我试着用几组不相等的size生成视频,都打不开。
解决方案:
(1)查看待原视频的size
(2)将代码里的size修改至相等,以上图为例,将size = (1920,1080)改为size = (1024,576)
----over