在跑背景替换的模型过程中,因为遇到ffmpeg的问题,所以就简单调了一下代码来实现。如何读取文件夹中某一类拥有相同后缀的图片,绕了一大圈,后来发现直接后缀.png就行。
import cv2
import glob
fps = 60 #保存视频的FPS,可以适当调整
fourcc = cv2.VideoWriter_fourcc(*'MJPG')
#saveVideo.avi是要生成的视频名称,(384,288)是图片尺寸
videoWriter = cv2.VideoWriter('Video.avi',fourcc,fps,(1920,1080))
#imge存放图片
imgs=glob.glob('output/*_matte.png')
for imgname in imgs:
frame = cv2.imread(imgname)
videoWriter.write(frame)
videoWriter.release()