Python视频转为GIF动画

时间:2022-05-23 程序员爱丽人气:0

Python 可用于读取常见的 MP4 视频格式并将其转换为 GIF动画。当然，如果你愿意，你可以使用预先构建的软件，但是自己做很有趣（并且是一种很好的学习体验）。

在本教程中，你将学习以下内容：

如何从 MP4 视频中提取帧
将帧转换为 GIF
创建 MP4 到 GIF GUI

让我们开始吧！

运行环境

你需要安装 OpenCV 绑定以读取 MP4 文件并将视频中的每一帧转换为 JPG 文件。安装教程：

python3 -m pip install opencv-python

你还需要Pillow从你从视频中提取的 JPG 创建动画 GIF。也可以用pip安装：

python3 -m pip install Pillow

要创建 GUI，我这里会用到PySimpleGUI。要安装该库，请使用以下命令：

python3 -m pip install PySimpleGUI

如果你使用的是 Anaconda，则包含 opencv-python 和 Pillow。你只需要单独安装 PySimpleGUI。

如何从 MP4 视频中提取帧

从 MP4 视频中提取帧的第一步是找到要转换为 GIF 的视频。

要从上面的视频中提取单个帧，你需要编写一些 Python。创建一个新文件并将其命名为mp4_converter.py。然后输入以下代码：

import cv2


def convert_mp4_to_jpgs(path):
    video_capture = cv2.VideoCapture(path)
    still_reading, image = video_capture.read()
    frame_count = 0
    while still_reading:
        cv2.imwrite(f"output/frame_{frame_count:03d}.jpg", image)
        
        # read next image
        still_reading, image = video_capture.read()
        frame_count += 1


if __name__ == "__main__":
    convert_mp4_to_jpgs("flask_demo.mp4")

此代码采用 MP4 视频文件的路径。然后使用cv2.VideoCapture(path)打开视频。你可以使用此方法通读整个视频并提取每一帧。提取帧时，可以使用cv2.imwrite()将其写出。

当你运行这段代码时，你会发现这个 7 秒的视频产生了 235 帧！

现在准备好拍摄这些帧并将它们转换为动画 GIF。

将帧变成 GIF

该过程的下一步是将使用 OpenCV 从 MP4 文件中提取的帧转换为动画 GIF。

这就是 Pillow 包的用武之地。你可以使用它来接收图像文件夹并创建你的 GIF。打开一个新文件并将其命名为gif_maker.py。然后输入以下代码：

import glob

from PIL import Image


def make_gif(frame_folder):
    images = glob.glob(f"{frame_folder}/*.jpg")
    images.sort()
    frames = [Image.open(image) for image in images]
    frame_one = frames[0]
    frame_one.save("flask_demo.gif", format="GIF", append_images=frames,
                   save_all=True, duration=50, loop=0)
    

if __name__ == "__main__":
    make_gif("output")

在这里，你使用 Python 的glob模块在输出文件夹中搜索 JPG 文件。然后对帧进行排序，使它们按正确的顺序排列。最后，你将它们保存为 GIF.

创建 MP4 到 GIF GUI

PySimpleGUI 是一个跨平台的 GUI 框架，可在 Linux、Mac 和 Windows 上运行。它封装了 Tkinter、wxPython、PyQt 和其他几个 GUI 工具包，为它们提供了一个通用接口。

在本文前面安装 PySimpleGUI 时，你安装了包装 Tkinter 的默认版本。

打开一个新的 Python 文件并将其命名为mp4_converter_gui.py。然后将此代码添加到你的文件中：

# mp4_converter_gui.py

import cv2
import glob
import os
import shutil
import PySimpleGUI as sg

from PIL import Image

file_types = [("MP4 (*.mp4)", "*.mp4"), ("All files (*.*)", "*.*")]


def convert_mp4_to_jpgs(path):
    video_capture = cv2.VideoCapture(path)
    still_reading, image = video_capture.read()
    frame_count = 0
    if os.path.exists("output"):
        # remove previous GIF frame files
        shutil.rmtree("output")
    try:
        os.mkdir("output")
    except IOError:
        sg.popup("Error occurred creating output folder")
        return
    
    while still_reading:
        cv2.imwrite(f"output/frame_{frame_count:05d}.jpg", image)
        
        # read next image
        still_reading, image = video_capture.read()
        frame_count += 1


def make_gif(gif_path, frame_folder="output"):
    images = glob.glob(f"{frame_folder}/*.jpg")
    images.sort()
    frames = [Image.open(image) for image in images]
    frame_one = frames[0]
    frame_one.save(gif_path, format="GIF", append_images=frames,
                   save_all=True, duration=50, loop=0)


def main():
    layout = [
        [
            sg.Text("MP4 File"),
            sg.Input(size=(25, 1), key="-FILENAME-", disabled=True),
            sg.FileBrowse(file_types=file_types),
        ],
        [
            sg.Text("GIF File Save Location"),
            sg.Input(size=(25, 1), key="-OUTPUTFILE-", disabled=True),
            sg.SaveAs(file_types=file_types),
            
        ],
        [sg.Button("Convert to GIF")],
    ]

    window = sg.Window("MP4 to GIF Converter", layout)

    while True:
        event, values = window.read()
        mp4_path = values["-FILENAME-"]
        gif_path = values["-OUTPUTFILE-"]
        if event == "Exit" or event == sg.WIN_CLOSED:
            break
        if event in ["Convert to GIF"]:
            if mp4_path and gif_path:
                convert_mp4_to_jpgs(mp4_path)
                make_gif(gif_path)
                sg.popup(f"GIF created: {gif_path}")

    window.close()


if __name__ == "__main__":
    main()

这是一段相当长的代码。我们来一段一段分析。

要开始，请查看导入部分：

# mp4_converter_gui.py

import cv2
import glob
import os
import shutil
import PySimpleGUI as sg

from PIL import Image

file_types = [("MP4 (*.mp4)", "*.mp4"), ("All files (*.*)", "*.*")]

在这里，你导入创建 GUI 应用程序所需的所有模块和包。这包括 OpenCV (cv2)、Pillow 的 Image clas 和 PySimpleGUI，以及许多 Python 自己的模块。

你还创建了一个变量来保存可以加载到 GUI 中的文件类型。这是一个元组列表。

现在是时候将注意力转向程序中的第一个函数：

def convert_mp4_to_jpgs(path):
    video_capture = cv2.VideoCapture(path)
    still_reading, image = video_capture.read()
    frame_count = 0
    if os.path.exists("output"):
        # remove previous GIF frame files
        shutil.rmtree("output")
    try:
        os.mkdir("output")
    except IOError:
        sg.popup("Error occurred creating output folder")
        return
    
    while still_reading:
        cv2.imwrite(f"output/frame_{frame_count:05d}.jpg", image)
        
        # read next image
        still_reading, image = video_capture.read()
        frame_count += 1

这是你之前创建的 MP4 转换器代码的修改版本。在这个版本中，你仍然使用VideoCapture()来读取 MP4 文件并将其转换为单独的帧。

但是，你还添加了一些额外的代码来删除“输出”文件夹（如果存在）。这可以防止你意外地将两个 MP4 文件合并到一个输出文件中，这会导致 GIF 混乱。

你还添加了一些代码来尝试在删除后创建“输出”文件夹。如果创建文件夹时出现错误，则会显示错误对话框。

其余代码与之前相同。

现在你已准备好查看下一个函数：

def make_gif(gif_path, frame_folder="output"):
    images = glob.glob(f"{frame_folder}/*.jpg")
    images.sort()
    frames = [Image.open(image) for image in images]
    frame_one = frames[0]
    frame_one.save(gif_path, format="GIF", append_images=frames,
                   save_all=True, duration=50, loop=0)

你可以使用make_gif()将帧文件夹转换为 GIF 文件。此代码与原始代码几乎相同，只是你传入 GIF 文件的路径以使其唯一。

最后一段代码是你的 GUI 代码：

def main():
    layout = [
        [
            sg.Text("MP4 File"),
            sg.Input(size=(25, 1), key="-FILENAME-", disabled=True),
            sg.FileBrowse(file_types=file_types),
        ],
        [
            sg.Text("GIF File Save Location"),
            sg.Input(size=(25, 1), key="-OUTPUTFILE-", disabled=True),
            sg.SaveAs(file_types=file_types),
            
        ],
        [sg.Button("Convert to GIF")],
    ]

    window = sg.Window("MP4 to GIF Converter", layout)

    while True:
        event, values = window.read()
        mp4_path = values["-FILENAME-"]
        gif_path = values["-OUTPUTFILE-"]
        if event == "Exit" or event == sg.WIN_CLOSED:
            break
        if event in ["Convert to GIF"]:
            if mp4_path and gif_path:
                convert_mp4_to_jpgs(mp4_path)
                make_gif(gif_path)
                sg.popup(f"GIF created: {gif_path}")

    window.close()


if __name__ == "__main__":
    main()

在 PySimpleGUI 中，当你想在用户界面中“布局”元素时，你可以将项目添加到 Python 列表中。对于此示例，你添加以下元素：

sg.Text - 此元素有两个实例。它们用作输入（文本框）的标签
sg.Input - 这个元素有两个实例，它是一个文本框类型的元素。一个保存 MP4 文件的位置，一个保存你要保存 GIF 的位置
sg.FileBrowse - 打开文件浏览对话框的按钮
sg.SaveAs - 打开文件另存为对话框的按钮
sg.Button - 一个可以做任何你想做的事情的按钮

接下来，你获取元素列表并将其传递给sg.Window，它表示包含所有其他元素的窗口。你的窗口还有一个退出按钮、一个最小化按钮和一个标题栏。

要启动 GUI 的事件循环，你需要创建一个 while 循环并从 Window 对象中读取数据。这允许你提取两个sg.Input()对象的值，其中包含 MP4 和 GIF 文件的路径。

当用户按下标记为“转换为 GIF”的按钮时，你会捕获该事件并调用convert_mp4_to_jpgs()和make_gif()。如果一切顺利，视频将被转换，你将看到一个弹出对话框，说明新创建的 GIF 的保存位置。

尝试运行此代码。你应该会看到如下内容：

现在你拥有了将 MP4 视频文件转换为 GIF 所需的所有内容。你可以采取多种不同的措施来改进你的代码。例如，你可以向代码中添加更多错误处理，以免意外覆盖 GIF。

你还可以添加一些新的 UI 元素来告诉你的代码调整各个帧的大小以帮助缩小 GIF。另一种选择是更改每个单独的 JPG 的压缩，这也将减小 GIF 的大小。

还有很多其他有趣的方法可以使这段代码变得更好。考虑一下，你一定会自己想出一些新功能！

加载全部内容