使用OpenCv禾嫉,對文件夾內(nèi)的視頻文件進(jìn)行分幀操作蚊丐,在VSCODE中可以直接運(yùn)行(更改為自己的文件夾地址),在Pycharm中改為自己的相對地址或者絕對地址孽椰,去掉r
import cv2
import os
import time
import datetime
import random
"""
Read video and convert it into pictures
讀取視頻并將視頻轉(zhuǎn)換為圖片
注意請勿在視頻文件夾中放其他文件
"""
def save_img():
# video_path = r'E:/VideoDir/'
# videos = os.listdir(video_path)
videos = load_video(r'E:/VideoDir/')
for video_name in videos:
vc = cv2.VideoCapture(video_name) # 讀入視頻文件
c = 1
if vc.isOpened(): # 判斷是否正常打開
rval, frame = vc.read()
else:
rval = False
timeF = 15 # 視頻幀計(jì)數(shù)間隔頻率
while rval: # 循環(huán)讀取視頻幀
rval, frame = vc.read()
if (c % timeF == 0): # 每隔timeF幀進(jìn)行存儲操作
if frame is not None:
cv2.imwrite("E:/EVideoDir2/" + str(random.randint(1, 100000)) + '_' + str(c) +
'.jpg', frame)
c = c + 1
cv2.waitKey(1)
vc.release()
def load_video(video_path):
input_path_extension = video_path.split('.')[-1]
if input_path_extension in ['mp4']:
return [video_path]
elif input_path_extension == "txt":
with open(video_path, "r") as f:
return f.read().splitlines()
else:
return glob.glob(os.path.join(video_path, "*.mp4"))
if __name__ == '__main__':
save_img()