网络编程
位置:首页>> 网络编程>> Python编程>> Python OpenCV实现视频分帧

Python OpenCV实现视频分帧

作者:等一杯咖啡  发布时间:2023-06-06 02:38:18 

标签:Python,OpenCV,视频分帧

本文实例为大家分享了Python OpenCV实现视频分帧的具体代码,供大家参考,具体内容如下


# coding=utf-8

import os
import cv2

videos_src_path = "/home/wgp/视频/"
video_formats = [".MP4", ".MOV"]
frames_save_path = "/home/wgp/视频/"
width = 320
height = 240
time_interval = 50

def video2frame(video_src_path, formats, frame_save_path, frame_width, frame_height, interval):
"""
将视频按固定间隔读取写入图片
:param video_src_path: 视频存放路径
:param formats:包含的所有视频格式
:param frame_save_path:保存路径
:param frame_width:保存帧宽
:param frame_height:保存帧高
:param interval:保存帧间隔
:return:帧图片
"""
videos = os.listdir(video_src_path)

def filter_format(x, all_formats):
if x[-4:] in all_formats:
 return True
else:
 return False

videos = filter(lambda x: filter_format(x, formats), videos)

for each_video in videos:
print "正在读取视频:", each_video

each_video_name = each_video[:-4]
os.mkdir(frame_save_path + each_video_name)
each_video_save_full_path = os.path.join(frame_save_path, each_video_name) + "/"

each_video_full_path = os.path.join(video_src_path, each_video)

cap = cv2.VideoCapture(each_video_full_path)
frame_index = 0
frame_count = 0
if cap.isOpened():
 success = True
else:
 success = False
 print("读取失败!")

while(success):
 success, frame = cap.read()
 print "---> 正在读取第%d帧:" % frame_index, success

if frame_index % interval == 0:
 resize_frame = cv2.resize(frame, (frame_width, frame_height), interpolation=cv2.INTER_AREA)
 # cv2.imwrite(each_video_save_full_path + each_video_name + "_%d.jpg" % frame_index, resize_frame)
 cv2.imwrite(each_video_save_full_path + "%d.jpg" % frame_count, resize_frame)
 frame_count += 1

frame_index += 1

cap.release()

if __name__ == '__main__':
video2frame(videos_src_path, video_formats, frames_save_path, width, height, time_interval)

来源:https://blog.csdn.net/bskfnvjtlyzmv867/article/details/79970146

0
投稿

猜你喜欢

手机版 网络编程 asp之家 www.aspxhome.com