import os
import cv2

# 要被合成的多張圖片所在文件夾
# 路徑分隔符最好使用“/”,而不是“\”,“\”本身有轉(zhuǎn)義的意思；或者“\\”也可以。
# 因?yàn)槭俏募A，所以最后還要有一個(gè)“/”
file_dir = 'C:/Users/YUXIAOYANG/Desktop/tset/'
list = []
for root ,dirs, files in os.walk(file_dir):
    for file in files:
        list.append(file)      # 獲取目錄下文件名列表

# VideoWriter是cv2庫提供的視頻保存方法，將合成的視頻保存到該路徑中
# 'MJPG'意思是支持jpg格式圖片
# fps = 5代表視頻的幀頻為5，如果圖片不多，幀頻最好設(shè)置的小一點(diǎn)
# (1280,720)是生成的視頻像素1280*720，一般要與所使用的圖片像素大小一致，否則生成的視頻無法播放
# 定義保存視頻目錄名稱和壓縮格式，像素為1280*720
video = cv2.VideoWriter('C:/Users/YUXIAOYANG/Desktop/test.mp4',cv2.VideoWriter_fourcc('m', 'p', '4', 'v'),5,(1981,991))

for i in range(1,len(list)):
    img = cv2.imread('C:/Users/YUXIAOYANG/Desktop/tset/'+list[i-1]) #讀取圖片
    print(img.shape)
    #img = cv2.resize(img,(1981,991)) #將圖片轉(zhuǎn)換為1280*720像素大小
    video.write(img) # 寫入視頻
    
# 釋放資源
video.release()

python圖片與視頻互轉(zhuǎn)（親測(cè)有效）

圖片轉(zhuǎn)視頻

1.任務(wù)需求背景

在標(biāo)注數(shù)據(jù)的過程中，需要【反復(fù)】瀏覽大量圖片（萬張以上的數(shù)量級(jí)），確認(rèn)圖片中的目標(biāo)類別以及室內(nèi)戶型布局。

但是，在電腦上瀏覽圖片有很大的不足：（a）需要持續(xù)點(diǎn)擊鼠標(biāo)或者鍵盤；（b）圖片加載跟不上點(diǎn)擊速度。

值得注意的是：網(wǎng)上有很多代碼（圖片轉(zhuǎn)視頻），但是真正能用的幾乎很少，本博文的代碼經(jīng)過測(cè)試，可以成功生成視頻。

2.代碼依賴庫

opencv-python==4.5.2
numpy==1.19.2
glob（python自帶模塊）

3.代碼實(shí)戰(zhàn)

基本步驟如下：

a. 使用glob獲取路徑下的所有圖片；
b. cv2.imread()讀取所有圖片；
c. 將讀取的圖片存儲(chǔ)在新的列表中，img_array；
d. 使用cv2.VideoWriter()創(chuàng)建VideoWriter對(duì)象，注意參數(shù)的設(shè)置；
e. 使用cv2.VideoWriter().write()保存 img_array 中的每一幀圖像到視頻文件；
f. 釋放 VideoWriter對(duì)象；

import cv2
import numpy as np
import glob
import os

# 其它格式的圖片也可以
img_array = []
for filename in glob.glob('E:/3DS1Data/20211118/29984CRL30V00067087/dataset/rgb/*.png'):
    img = cv2.imread(filename)
    height, width, layers = img.shape
    size = (width, height)
    img_array.append(img)

# avi：視頻類型，mp4也可以
# cv2.VideoWriter_fourcc(*'DIVX')：編碼格式
# 5：視頻幀率
# size:視頻中圖片大小
out = cv2.VideoWriter('E:/3DS1Data/20211118/29984CRL30V00067087/dataset/project-all.avi',
                      cv2.VideoWriter_fourcc(*'DIVX'),
                      5, size)

for i in range(len(img_array)):
    out.write(img_array[i])
out.release()

下圖是原始圖片數(shù)據(jù)：