视频转图片 python代码
最近拍摄在拍摄近红外rgb的数据集,需要将相关的视频转换为图片。参考了一些博客,更改了一些错误。代码如下
#! encoding: UTF-8
import os
import cv2
import cv
videos_src_path = '/home/zhutianhao/mp4/'
videos_save_path = '/home/zhutianhao/jpg'
videos = os.listdir(videos_src_path)
videos = filter(lambda x: x.endswith('mp4'), videos)
for each_video in videos:
print('Video Name :',each_video)
#获取每一个视频的名字,并创建对于的文件夹存储
each_video_name, _ = each_video.split('.')
os.mkdir(videos_save_path + '/' + each_video_name)
each_video_save_full_path = os.path.join(videos_save_path, each_video_name) + '/'
each_video_full_path = os.path.join(videos_src_path, each_video)
cap = cv2.VideoCapture(each_video_full_path)
frame_count = 1
success = True
while(success):
success, frame = cap.read()
i