代码功能为,对于视频xxx.mp4,将其中第39 ~ 第105秒,第120 ~ 第160秒的两个片段逐帧转为图片,并分别存入两个文件夹中。
转换的视频及结果如下:
原始视频:
转换结果:
每个文件夹内:
代码如下:
import cv2
import os
video_path = "xxx.mp4" # 插入视频路径
starts_time = [39, 120] # 截取视频的时间,此处为第39~第105秒,第120~第160秒
ends_time = [105, 160]
starts = [24 * x for x in starts_time] # 计算起止帧的idx,其中,24为帧率
ends = [24 * x for x in ends_time]
cap = cv2.VideoCapture(video_path)
num = 0
success = True
for i in range(len(starts)):
save_dir = os.path.join(video_path[:-4] + str(i) + '_' + str(starts_time[i]) + '_' + str(ends_time[i]))
if not os.path.exists(save_dir):
os.mkdir(save_dir)
sub_num = 0
while(success and num < ends[i]):
success, frame = cap.read()
num += 1
if success and num >= starts[i]:
save_name = '{:0>5d}.png'.format(sub_num)
save_path = os.path.join(save_dir, save_name)
cv2.imwrite(save_path, frame)
print('Image {} saved in {}.'.format(save_name, save_dir))
sub_num += 1
cap.release()