在 IPython/Jupyter 笔记本中读取/写入/显示图像和视频。

[GitHub 源码]   [API 文档]   [Colab 示例]

示例：

查看笔记本   mediapy_examples.ipynb      

图像：

    !pip install -q mediapy
    import mediapy as media
    import numpy as np

    image = media.read_image('https://github.com/hhoppe/data/raw/main/image.png')
    print(image.shape, image.dtype)  # 这是一个 numpy 数组。
    media.show_image(image)

    checkerboard = np.kron([[0, 1] * 16, [1, 0] * 16] * 16, np.ones((4, 4)))
    media.show_image(checkerboard)

    media.show_image(media.color_ramp((128, 128)), height=48, title='ramp')

    images = {
        'original': image,
        'brightened': media.to_float01(image) * 1.5,
    }
    media.show_images(images)

    media.write_image('/tmp/checkerboard.png', checkerboard)

视频：

    url = 'https://github.com/hhoppe/data/raw/main/video.mp4'
    video = media.read_video(url)
    print(video.shape, video.dtype)  # 这是一个 numpy 数组。
    print(video.metadata.fps)  # 'metadata' 属性包含帧率。
    media.show_video(video)  # 使用获取的帧率播放视频。

    media.show_images(video, height=80, columns=4)  # 并排显示帧。

    video = media.moving_circle((128, 128), num_images=10)
    media.show_video(video, fps=10)

    media.write_video('/tmp/video.mp4', video, fps=60)

    # 逐帧降低视频亮度：
    filename_in = '/tmp/video.mp4'
    filename_out = '/tmp/out.mp4'
    with media.VideoReader(filename_in) as r:
      print(f'shape={r.shape} fps={r.fps} bps={r.bps}')
      darken_image = lambda image: media.to_float01(image) * 0.5
      with media.VideoWriter(
          filename_out, shape=r.shape, fps=r.fps, bps=r.bps) as w:
        for image in r:
          w.add_image(darken_image(image))
    media.show_video(media.read_video(filename_out), fps=60)

设置：

视频 I/O 依赖于外部程序 ffmpeg，必须在系统 PATH 中存在。在 Unix 上，可以使用以下命令安装：

    apt install ffmpeg

或在笔记本中使用：

    !command -v ffmpeg >/dev/null || (apt update && apt install -y ffmpeg)