File size: 1,692 Bytes
d39b279 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 |
import os
from glob import glob
from moviepy.editor import VideoFileClip
import multiprocessing
import cv2, math
def get_video_length(file_path):
video = VideoFileClip(file_path)
return video.duration
def process_video(video_path):
video_name = video_path.split('/')[-1]
video_name = video_name.split('.')[:-1]
video_name = '.'.join(video_name)
path = video_path.split('/')[1:-1]
path = '/'.join(path)
image_path = './SD_frames/'+path+'/'+ video_name+'/'
print(video_name)
if os.path.exists(image_path):
print("路径存在")
else:
print(video_name, "路径不存在")
try:
try:
video_length = get_video_length(video_path)
print(video_name, f"视频长度为：{video_length} 秒")
os.makedirs(os.path.dirname(image_path), exist_ok=True)
if video_length >= 4 :
inter_val = 2
os.system(f"cd {image_path} | ffmpeg -loglevel quiet -i {video_path} -r {inter_val} {image_path}%d.jpg")
else:
inter_val = math.ceil(8 / video_length)
os.system(f"cd {image_path} | ffmpeg -loglevel quiet -i {video_path} -r {inter_val} {image_path}%d.jpg")
except Exception as e:
print("发生异常：", str(e))
except:
print("Skip")
if __name__ == '__main__':
print("Getting frames!!")
video_paths = './SD'
all_dirs = []
all_dirs = glob(video_paths+'/*')
print(all_dirs)
pool = multiprocessing.Pool(processes=8)
pool.map(process_video, all_dirs)
pool.close()
pool.join()
|