使用百度PaddleHub制作抠图视频

目标：制作在太空中跳街舞的视频

import paddlehub as hub
from PIL import Image
import os
import cv2
import matplotlib.pyplot as plt   # 显示图片
import matplotlib.image as mpimg  # 读取图片
import numpy as np

第一步：

def video2jpg(video_file,output_path):
    '''
    将视频文件video_file每一帧转成图片保存到output_path文件夹
    '''
    try:
        os.makedirs(output_path) #创建输出文件夹
    except:
        print()

    #读取视频文件
    cap = cv2.VideoCapture(video_file)
    
    num = 0
    while(True):
        ret,frame = cap.read() 
        if ret:
            cv2.imwrite('%s%d.jpg'%(output_path,num), frame)
            num += 1
        else:
            break
    cap.release()#关闭视频

#执行该步子程序
video_file='video/1.mp4'
output_path='video/video2jpg/'
video2jpg(video_file,output_path)

!ls video

第二步：

%matplotlib inline

def humanseg(images,output_dir):
    ''