发布于2019-09-11 13:07 阅读(971) 评论(0) 点赞(20) 收藏(3)
本文是基于opencv将视频和动态图gif保存为图像帧。可以根据输入视频格式的不同,修改第21行。
对动图的处理不同于视频,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
作者:倒车请注意
链接:https://www.pythonheidong.com/blog/article/107105/6d31438fda44cd56afb0/
来源:python黑洞网
任何形式的转载都请注明出处,如有侵权 一经发现 必将追究其法律责任
昵称:
评论内容:(最多支持255个字符)
---无人问津也好,技不如人也罢,你都要试着安静下来,去做自己该做的事,而不是让内心的烦躁、焦虑,坏掉你本来就不多的热情和定力
Copyright © 2018-2021 python黑洞网 All Rights Reserved 版权所有,并保留所有权利。 京ICP备18063182号-1
投诉与举报,广告合作请联系vgs_info@163.com或QQ3083709327
免责声明:网站文章均由用户上传,仅供读者学习交流使用,禁止用做商业用途。若文章涉及色情,反动,侵权等违法信息,请向我们举报,一经核实我们会立即删除!