【jupyter显示视频】

一、前言

• 使用jupyter的时候，并使用不了cv库对图片/视频进行显示，不方便图像处理效果的观察。一般图片采用的是matplotlib库对图片进行显示，视频的话则采用HTML的形式进行展示。下面简单介绍使用教程。

二、安装库&附加练习视频

pip install opencv-python -i https://pypi.douban.com/simple
pip install imageio -i https://pypi.douban.com/simple
pip install imageio-ffmpeg -i https://pypi.douban.com/simple
pip install matplotlib -i https://pypi.douban.com/simple
pip install IPython -i https://pypi.douban.com/simple

# 或者使用下面语句，利用豆瓣源，快速安装多个库
pip install --index-url https://pypi.douban.com/simple imageio imageio-ffmpeg matplotlib IPython

三、显示代码

# 导入相关包
import cv2
# 导入一些可视化需要的包
import imageio
import matplotlib.animation as animation
from IPython.display import HTML

# 定义一个展示视频的函数
def display(driving, fps, size=(8, 6)):
    fig = plt.figure(figsize=size)

    ims = []
    for i in range(len(driving)):
        cols = []
        cols.append(driving[i])

        im = plt.imshow(np.concatenate(cols, axis=1), animated=True)
        plt.axis('off')
        ims.append([im])

    video = animation.ArtistAnimation(fig, ims, interval=1000.0/fps, repeat_delay=1000)

    plt.close()
    return video



# 展示一下输入的视频, 如果视频太大，时间会非常久，可以跳过这个步骤
video_path = '2.mp4'
video_frames = imageio.mimread(video_path, memtest=False)

# 获得视频的原分辨率
cap = cv2.VideoCapture(video_path)
fps = cap.get(cv2.CAP_PROP_FPS)
    
HTML(display(video_frames, fps).to_html5_video())

四、效果

五、参考链接

飞浆参考文章