十年网站开发经验 + 多家企业客户 + 靠谱的建站团队
量身定制 + 运营维护+专业推广+无忧售后,网站问题一站解决
本文是基于opencv将视频和动态图gif保存为图像帧。可以根据输入视频格式的不同,修改第21行。
下陆ssl适用于网站、小程序/APP、API接口等需要进行数据传输应用场景,ssl证书未来市场广阔!成为创新互联的ssl证书销售渠道,可以享受市场价格4-6折优惠!如果有意向欢迎电话联系或者加微信:13518219792(备注:SSL证书合作)期待与您的合作!对动图的处理不同于视频,PIL库包含对图像序列的基本支持。当打开gif图像时,自动加载第一帧。当图像读取完成时,抛出EOFError异常。我们可以使用seek()与tell()函数完成图像帧的读取。
本代码的前部分是对文件的读取。数据集文件结构如下:
|——datasets |——action1 action1_1.gif action1_2.gif ...... |——action2 action2_1.gif action2_2.gif ...... import cv2 import os from PIL import Image video_path = 'PATH_ROOT/datasets/' #视频或gif图像的路径 save_path = 'PATH_ROOT/save/' #保存帧的路径 action_list = os.listdir(video_path) for action in action_list: if not os.path.exists(save_path+action): os.mkdir(save_path+action) video_list = os.listdir(video_path+action) for video in video_list: prefix = video.split('.')[0] if not os.path.exists(save_path+action+'/'+prefix): os.mkdir(save_path+action+'/'+prefix) save_name = save_path + action + '/' + prefix + '/' video_name = video_path+action+'/'+video name = video_name.split('.')[1] if name == "avi": cap = cv2.VideoCapture(video_name) fps = int(cap.get(cv2.CAP_PROP_FRAME_COUNT)) fps_count = 0 for i in range(fps): ret, frame = cap.read() if ret: cv2.imwrite(save_name + str(10000 + fps_count) + '.jpg', frame) fps_count += 1 if name == "gif": im = Image.open(video_name) #当打开一个序列文件时,PIL库自动加载第一帧。 #可以使用seek()函数和tell()函数在不同帧之间移动。实现保存 try: while True: current = im.tell() #为了保存为jpg格式,需要转化。否则可以保存为png img = im.convert('RGB') img.save(save_name+'/'+str(10000+current)+'.jpg') im.seek(current + 1) except EOFError: pass