python将照片集导出成视频

shigen坚持更新文章的博客写手,记录成长,分享认知,留住感动。个人IP:shigen

背景

一个安静的下午,看着电脑里乱七八糟的照片,有大有小,宽高不一,突然想找个方式把他们统一起来,然后做成视频更好(其实我在上高中的时候就喜欢把照片做成视频,觉得意义很大)。要满足批量、自动化,肯定得动用代码了。于是首先我列举了下我希望的功能:

  1. 照片来源:制定的目录下所有的文件格式为照片的文件,按照照片的文件名进行排序
  2. 照片质量:按照目前的720P、1080P、甚至是2K、4K的画质来生成照片视频
  3. 照片横竖:可以自定义指定照片是横屏还是竖屏
  4. 照片时间:可以自定义每一张照片的放映时间
  5. 照片比例:这个也是最为重要的,对此我分成了如下几种case:
    1. 图片的宽度 < 视频宽度 * 50% or 图片的高度 < 视频高度 * 50%:舍弃掉
    2. 图片宽度 < 视频宽度 or 图片的高度 < 视频高度:居中等比放大,直到高度 = 视频高度 or 宽度 = 视频宽度=
    3. 其它尺寸,图片居中等比缩小,直到高度 = 视频高度 or 宽度 = 视频宽度

这种照片比例的放大居中,基本上是强迫症患者的福音了,严格的居中对齐。

实现

依托强大的python库,这里主要用到的工具库有:

  1. Pillow,源代码有简单、精炼的解释:**Pillow is the friendly PIL fork by Alex Clark and Contributors.PIL is the Python Imaging Library by Fredrik Lundh and Contributors.**机器学习、图像识别等场景用到的最多,这里主要使用它来调整图片的大小
  2. MoviePy, 是一个用于视频编辑的 Python 库。可以剪辑视频添加音频和字幕调整视频帧简单的特效等等。这里主要是根据照片序列生成视频。

介绍完主要的库之后,就是代码环节了,代码里注释较多,轻松入手,不做过多的解释:

python 复制代码
# create viode from a dictionary which contains image or video files
import os
from PIL import Image
from moviepy.editor import ImageSequenceClip


def resize_and_crop(image, target_size):
    """将图片根据给定的大小进行缩放和裁剪

    1. 图片的宽度 < 视频宽度 * 50% or 图片的高度 < 视频高度 * 50%:舍弃掉
    2. 图片宽度 < 视频宽度 or 图片的高度 < 视频高度:居中等比放大,直到高度 = 视频高度 or 宽度 = 视频宽度
    3. 图片居中等比缩小,直到高度 = 视频高度 or 宽度 = 视频宽度

    Args:
        image (str): 原图片文件路径
        target_size (tuple): 视频大小(宽度, 高度)

    Returns:
        Image: 调整后的图片,可能为空
    """
    img = Image.open(image)
    video_width, video_height = target_size

    # 检查条件1:如果原图宽度或高度小于视频尺寸的50%,则返回None
    if img.width < video_width * 0.5 or img.height < video_height * 0.5:
        return None

    # 计算目标缩放比例
    scale_x = video_width / img.width
    scale_y = video_height / img.height

    # 根据需要的宽度和高度选择缩放比例
    if img.width < video_width or img.height < video_height:
        # 居中等比放大
        scale = max(scale_x, scale_y)
    else:
        # 居中等比缩小
        scale = min(scale_x, scale_y)

    # 放大或缩小图片
    new_size = (int(img.width * scale), int(img.height * scale))
    img = img.resize(new_size, Image.ANTIALIAS)

    # 计算裁剪框的位置
    left = (img.width - video_width) // 2
    top = (img.height - video_height) // 2
    right = left + video_width
    bottom = top + video_height

    # 裁剪并返回最终图片
    img = img.crop((left, top, right, bottom))
    return img


def create_video_from_images(folder_path,
                             out_put_file_name='output_video.mp4',
                             resolution='720p',
                             is_horizontal=True,
                             duration=3):
    """通过照片生成视频

    Args:
        folder_path (_type_): 文件夹路径
        out_put_file_name (str, optional): _description_. 视频输出路径 Defaults to 'output_video.mp4'.
        resolution (str, optional): _description_. 视频清晰度 Defaults to '720p'.
        is_horizontal (bool, optional): _description_. 是否是横屏 Defaults to True.
        duration (int, optional): _description_. 每张照片的放映时长 Defaults to 3.

    Raises:
        ValueError: 视频清晰度错误
    """
    resolution_mapping = {
        '720p': (1280, 720),
        '1080p': (1920, 1080),
        '2k': (2560, 1440),
        '4k': (3840, 2160),
    }

    if resolution not in resolution_mapping:
        raise ValueError(
            "Invalid resolution. Choose from '720p', '1080p', '2k', '4k'.")

    target_size = resolution_mapping[resolution]
    if not is_horizontal:
        target_size = (target_size[1], target_size[0])

    images = []

    # 读取文件夹下的文件并按照文件名排序
    for filename in sorted(os.listdir(folder_path)):
        if filename.lower().endswith(('.png', '.jpg', '.jpeg', '.bmp')):
            img = resize_and_crop(os.path.join(folder_path, filename),
                                  target_size)
            if img:
                images.append(img)

    # 照片的临时位置
    temp_files = []
    for i, img in enumerate(images):
        temp_file = f'temp_image_{i}.png'
        img.save(temp_file)
        temp_files.append(temp_file)

    # 创建视频
    clip = ImageSequenceClip(temp_files, fps=1 / duration)
    output_file = os.path.join(folder_path, out_put_file_name)
    clip.write_videofile(output_file, codec='libx264')

    # 清除临时文件
    for temp_file in temp_files:
        os.remove(temp_file)


if __name__ == '__main__':
    create_video_from_images(
        folder_path='/Users/xxxx/Downloads/xxx/imgs',
        is_horizontal=False,
        resolution='720p')

我们执行脚本,这里是控制台输出:

再来看看视频输出:

标准的720P H264编码,3&,21s的时长。有一张图是横屏的图,这里生成的视频中也根据高度放大进行了居中裁剪:

整体的感觉还不错,特此写个博客分享出来。当然还有很多的优化点:

优化项

其实做的还是相当的粗糙,但是基本上还是省事儿了。考虑到的优化点有:

  1. 指定背景音乐并实现照片卡点
  2. 加上随机的特效切换

咳,目前想到的就这么多。作为工具,我觉得越简单越好,需要效率和我开发时间的权衡。

与shigen一起,每天不一样!

相关推荐
华清远见IT开放实验室3 分钟前
【每天学点AI】实战图像增强技术在人工智能图像处理中的应用
图像处理·人工智能·python·opencv·计算机视觉
界面开发小八哥7 分钟前
更高效的Java 23开发,IntelliJ IDEA助力全面升级
java·开发语言·ide·intellij-idea·开发工具
mqiqe28 分钟前
Elasticsearch 分词器
python·elasticsearch
qystca35 分钟前
洛谷 B3637 最长上升子序列 C语言 记忆化搜索->‘正序‘dp
c语言·开发语言·算法
薯条不要番茄酱36 分钟前
数据结构-8.Java. 七大排序算法(中篇)
java·开发语言·数据结构·后端·算法·排序算法·intellij-idea
runing_an_min40 分钟前
ffmpeg视频滤镜:替换部分帧-freezeframes
ffmpeg·音视频·freezeframes
今天吃饺子41 分钟前
2024年SCI一区最新改进优化算法——四参数自适应生长优化器,MATLAB代码免费获取...
开发语言·算法·matlab
努力进修1 小时前
“探索Java List的无限可能:从基础到高级应用“
java·开发语言·list
不去幼儿园2 小时前
【MARL】深入理解多智能体近端策略优化(MAPPO)算法与调参
人工智能·python·算法·机器学习·强化学习
Ajiang28247353043 小时前
对于C++中stack和queue的认识以及priority_queue的模拟实现
开发语言·c++