快捷導(dǎo)航

Python將視頻或者動態(tài)圖gif逐幀保存為圖片的方法

更新時間：2019年09月10日 09:37:22 作者：小宇少俠

本文是基于opencv將視頻和動態(tài)圖gif保存為圖像幀的方法，本文通過實例代碼給大家介紹的非常詳細，具有一定的參考借鑒價值，需要的朋友參考下吧

本文是基于opencv將視頻和動態(tài)圖gif保存為圖像幀。可以根據(jù)輸入視頻格式的不同，修改第21行。

對動圖的處理不同于視頻，PIL庫包含對圖像序列的基本支持。當(dāng)打開gif圖像時，自動加載第一幀。當(dāng)圖像讀取完成時，拋出EOFError異常。我們可以使用seek()與tell()函數(shù)完成圖像幀的讀取。

本代碼的前部分是對文件的讀取。數(shù)據(jù)集文件結(jié)構(gòu)如下：

|——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)
   #當(dāng)打開一個序列文件時，PIL庫自動加載第一幀。
   #可以使用seek()函數(shù)和tell()函數(shù)在不同幀之間移動。實現(xiàn)保存
   try:
    while True:
     current = im.tell()
     #為了保存為jpg格式，需要轉(zhuǎn)化。否則可以保存為png
     img = im.convert('RGB') 
     img.save(save_name+'/'+str(10000+current)+'.jpg')
     im.seek(current + 1)
   except EOFError:
    pass

知識點擴展：python如何將gif動態(tài)圖轉(zhuǎn)化成每幀圖片

from PIL import Image
import os
gifFileName = 'test.gif'
#使用Image模塊的open()方法打開gif動態(tài)圖像時，默認是第一幀
im = Image.open(gifFileName)
pngDir = gifFileName[:-4]
#創(chuàng)建存放每幀圖片的文件夾
os.mkdir(pngDir)
try:
 while True:
  #保存當(dāng)前幀圖片
  current = im.tell()
  im.save(pngDir+'/'+str(current)+'.png')
  #獲取下一幀圖片
  im.seek(current+1)
except EOFError:
  pass

總結(jié)

以上所述是小編給大家介紹的Python將視頻或者動態(tài)圖gif逐幀保存為圖片的方法,希望對大家有所幫助，如果大家有任何疑問請給我留言，小編會及時回復(fù)大家的。在此也非常感謝大家對腳本之家網(wǎng)站的支持！
如果你覺得本文對你有幫助，歡迎轉(zhuǎn)載，煩請注明出處，謝謝！

您可能感興趣的文章: