Python實現圖片和視頻的相互轉換

Posted on by

使用背景

有時候我們需要把很多的圖片合成視頻，或者說自己寫一個腳本去加快或者放慢視頻；

也有時候需要把視頻裁剪成圖片，進行後續操作。

這裡提供兩種方法，一是視頻轉圖像；二是圖像轉視頻。 

一、視頻轉圖像

有時候我們需要把文件夾中的視頻按照一定的幀率截取圖片，如一秒取三張，為瞭實現這一需求，我特地編寫瞭代碼實現，並且附上瞭十分詳細的說明，為瞭方便大傢改代碼實現自己需求（主要為瞭照顧剛學python的或者隻需要用這一需求的小夥伴），本博文的特點如下：

  • 直接改變輸入文件夾和輸出文件夾的位置，就可以實現功能
  • 輸出的圖片能按照一定格式命名，這裡是以20210823_0001命名
  • 逐一遍歷文件夾中視頻，序號之間可以連續（也可不連續，需要改代碼）
  • 利用雙線性插值方法，就算圖片變大瞭，也能保證分辨率（關於雙線性，這裡隻是調用opencv方法實現，具體實現代碼和原理可看我下一篇博文）
  • 代碼解釋十分詳細，一看就懂

十分詳細代碼實現

首先先說明需要自己修改的參數，代碼如下：

filepath = 'C:/Users/ZFG/Desktop/1111' #視頻文件夾所在目錄
data='20210823'   #要是儲存的文件為20210823_0001格式，則為前半部分
save_filename='C:/Users/ZFG/Desktop/2222/'  #儲存圖片的文件夾的地址
timeF = 6   #根據一秒取多少幀設置，比如我的視頻是24幀/秒，取6則一秒取三張
kernal=(700,700)  #設置輸出的大小，根據自己需求設置

然後再設置一個方法，來儲存截取後的圖片：

def saveImage(image,SaveAddress,num):  #image為讀取的圖片，SaveAddress為需要存的地址，num為截取圖片時候記錄的序號
    address= SaveAddress+data+'_'+str(num).zfill(4)+'.jpg'   #這裡設置輸出格式
    cv2.imwrite(address,image) #這裡為存圖片

之後讀取文件夾：

pathDir = os.listdir(filepath)

i=0
j=0
for allDir in pathDir:  #遍歷文件夾中的每一個視頻
    videopath =filepath+'/'+ allDir
    videoCapture=cv2.VideoCapture(videopath)  #輸入絕對路徑
    untill,picture=videoCapture.read()  #讀取視頻，視頻讀取完的時候，返回的untill為False，表示視頻讀取完畢
    while untill:
        i+=1
        if (i%timeF==0):    #根據原視頻的幀率看截圖圖片的頻率
            j+=1
            picture=cv2.resize(picture,kernal,cv2.INTER_LINEAR)  #這裡調用瞭opencv中的雙線性插值法，要是圖片增加很快，保證瞭圖片精度
            saveImage(picture,save_filename,j)  #調用我們之前描述的方法
            pass
        untill, picture = videoCapture.read()  #再次看視頻是否結束，結束瞭則until為False

將上訴代碼結合後，最終程序如下所示：

import cv2
import os

i = 0
j = 0
pathDir = os.listdir(filepath)
filepath = 'C:/Users/ZFG/Desktop/1111'
data='20210823'
save_filename='C:/Users/ZFG/Desktop/2222/'
timeF = 6
kernal=(700,700)
def saveImage(image,SaveAddress,num):
    address= SaveAddress+data+'_'+str(num).zfill(4)+'.jpg'
    cv2.imwrite(address,image)
for allDir in pathDir:
    videopath =filepath+'/'+ allDir
    videoCapture=cv2.VideoCapture(videopath)
    untill,picture=videoCapture.read()
    while untill:
        i+=1
        if (i%timeF==0):
            j+=1
            picture=cv2.resize(picture,kernal,cv2.INTER_LINEAR)
            saveImage(picture,save_filename,j)
            pass
        untill, picture = videoCapture.read()

最後看下效果吧：

第一張圖是文件夾中的視頻，幀率是24幀/秒，第二張圖是一秒取三張圖片後，圖片儲存在文件夾中的圖。

二、圖像轉視頻

有時候我們需要把很多的圖片合成視頻，或者說自己寫一個腳本去加快或者放慢視頻。我寫這個的目的，是因為我有一個模型隻能處理圖片，但是我想看視頻的處理效果，於是我先視頻變成圖片，然後處理好後把圖片變成視頻，這樣就解決需求啦~

十分詳細代碼實現

首先先說明需要自己修改的參數，代碼如下：

if __name__ == '__main__':
    im_dir = ' '  # 圖片幀存放路徑，這裡寫一個文件夾
    dir_list=os.listdir(im_dir)
    fps = 20  # 設置一個幀率，每秒鐘幀數越多，視頻就越快
    dir_video=' ' #合成後視頻的存放視頻
    video_dir = dir_video + '.avi'
    frame2video(im_dir, video_dir, fps)

然後再設置一個方法，來儲存截取後的圖片：

def frame2video(im_dir, video_dir, fps):
    im_list = os.listdir(im_dir)
    im_list.sort(key=lambda x: int(x.replace("frame", "").split('.')[0])) 
    img = Image.open(os.path.join(im_dir, im_list[0]))
    img_size = img.size  # 獲得圖片分辨率，文件夾下的圖片分辨率需要一致，要是不一致可以寫一個if函數resize一下~
    fourcc = cv2.VideoWriter_fourcc(*'XVID')  # opencv版本是3
    videoWriter = cv2.VideoWriter(video_dir, fourcc, fps, img_size)
    for i in im_list:
        im_name = os.path.join(im_dir + i)
        frame = cv2.imdecode(np.fromfile(im_name, dtype=np.uint8), -1)
        videoWriter.write(frame)
    videoWriter.release()
    print('Done')

大功告成

import cv2
import os
import numpy as np
from PIL import Image

def frame2video(im_dir, video_dir, fps):
    im_list = os.listdir(im_dir)
    im_list.sort(key=lambda x: int(x.replace("frame", "").split('.')[0]))
    img = Image.open(os.path.join(im_dir, im_list[0]))
    img_size = img.size 
    fourcc = cv2.VideoWriter_fourcc(*'XVID')  # opencv版本是3
    videoWriter = cv2.VideoWriter(video_dir, fourcc, fps, img_size)

    for i in im_list:
        im_name = os.path.join(im_dir + i)
        frame = cv2.imdecode(np.fromfile(im_name, dtype=np.uint8), -1)
        videoWriter.write(frame)

    videoWriter.release()
    print('Done')


if __name__ == '__main__':
    im_dir = ' '
    dir_list=os.listdir(im_dir)
    fps = 20
    dir_video=' '
    video_dir = dir_video + '.avi'
    frame2video(im_dir, video_dir, fps)

以上就是Python實現圖片和視頻的相互轉換的詳細內容，更多關於Python圖片和視頻相互轉換的資料請關註WalkonNet其它相關文章！

推薦閱讀：