def frame_process(vis):
input_path, video_name = vis
out_path = os.path.join(input_path, 'frame')
image_path = os.path.join(out_path, video_name[:-4])
if not os.path.exists(image_path):
os.makedirs(image_path)
im_num = 0
cap = cv2.VideoCapture(os.path.join(input_path, video_name))
while True:
ret, frame = cap.read()
if ret:
cv2.imwrite(image_path + '/' + '{:06d}.jpg'.format(im_num), frame)
im_num += 1
else:
break
if os.path.isfile(root_path):
file_list = [root_path]
else:
file_list = os.listdir(root_path)
vis_list = []
for i in file_list:
if not i.endswith('mp4'):
continue
if os.path.isfile(i):
data = os.path.split(i)
vis_list.append(data)
break
data = (file_list, i)
vis_list.append(data)
for vis in vis_list:
print(vis)
frame_process(vis)
import os
网友评论