python中将多张图片合成为视频

﹏ヽ暗。殇╰゛Y 2023-01-05 12:43 192阅读 0赞

转自：[https://blog.csdn.net/choumin/article/details/90319477][https_blog.csdn.net_choumin_article_details_90319477]

很多程序示例描述了如何在python中将多张图片合成为视频，但好像都没有关注一个小问题——被合成的那些图片的尺寸应该一致，并且和程序中给定的参数一致。我遇到的问题就是，如果多张图片的尺寸不一致，或者如果图片的尺寸与给定的参数不一致，则生成出来的视频会有问题，无法播放。针对这个问题，有必要对多张图片的尺寸进行统一，即通过下面的resize函数实现。

import cv2
    import glob
     
    def resize(img_array, align_mode):
        _height = len(img_array[0])
        _width = len(img_array[0][0])
        for i in range(1, len(img_array)):
            img = img_array[i]
            height = len(img)
            width = len(img[0])
            if align_mode == 'smallest':
                if height < _height:
                    _height = height
                if width < _width:
                    _width = width
            else:
                if height > _height:
                    _height = height
                if width > _width:
                    _width = width
     
        for i in range(0, len(img_array)):
            img1 = cv2.resize(img_array[i], (_width, _height), interpolation=cv2.INTER_CUBIC)
            img_array[i] = img1
     
        return img_array, (_width, _height)
     
    def images_to_video(path):
        img_array = []
     
        for filename in glob.glob(path+'/*.png'):
            img = cv2.imread(filename)
            if img is None:
                print(filename + " is error!")
                continue
            img_array.append(img)
     
        # 图片的大小需要一致
        img_array, size = resize(img_array, 'largest')
        fps = 1
        out = cv2.VideoWriter('demo.avi', cv2.VideoWriter_fourcc(*'DIVX'), fps, size)
     
        for i in range(len(img_array)):
            out.write(img_array[i])
        out.release()
     
    def main():
        path = "../pictures/test/"
        images_to_video(path)
     
    if __name__ == "__main__":
        main()

[https_blog.csdn.net_choumin_article_details_90319477]: https://blog.csdn.net/choumin/article/details/90319477