技术路线速通！用飞桨让京剧人物照片动起来

Python#安装paddlehub及其模型!pip install paddlehub# 下载模型%%capture!hub install stylepro_artistic==1.0.1

Pythonimport paddlehub as hubfrom PIL import Imageimport numpy as npimport matplotlib.pyplot as plt
# 设置中文字体和编码#plt.rcParams['font.sans-serif']=['Microsoft YaHei']

#输入源图片与风格图片picture = '/home/aistudio/work/image_source.jpg'style_image = '/home/aistudio/work/style.png'
#选择模型
stylepro_artistic = hub.Module(name="stylepro_artistic")

content_image = Image.open(picture).convert('RGB')style_image = Image.open(style_image).convert('RGB')content_image = np.array(content_image)style_image = np.array(style_image)
# 进行风格迁移images = [    {        'content': content_image,  # 内容图像        'styles'




    
: [style_image],   # 风格图像        'weights': [0.3]           # 风格图像的权重    }]
result = stylepro_artistic.style_transfer(    images=images,    visualization=True)

!mv ./transfer_result/* ./transfer_result.jpg!rm -rf ./transfer_result!mv ./transfer_result.jpg ./output/transfer_result.jpg
fig, axes = plt.subplots(1, 3, figsize=(12, 4))
# 图像文件路径image_paths = [    '/home/aistudio/work/image_source.jpg',    '/home/aistudio/work/style.png',    './output/transfer_result.jpg']
titles = [    'Original Image',    'Style Reference Image',    'Output Result']
for i, ax in enumerate(axes):    ax.set_title(titles[i])    ax.axis('off')    ax.imshow(plt.imread(image_paths[i]))
print("图片保存路径为/home/aistudio/output/transfer_result.jpg")

Python# 从github上克隆PaddleGAN代码# 从github上克隆PaddleGAN代码（如下载速度过慢，可用gitee源）!git clone https://gitee.com/paddlepaddle/PaddleGAN#!git clone https://github.com/PaddlePaddle/PaddleGAN
# 安装所需安装包%cd PaddleGAN/!pip install -r requirements.txt!pip install imageio-ffmpeg!pip install ppgan

Python# 生成的结果保存的路径如下 ---> /home/aistudio/PaddleGAN/applications/output/result.mp4
%cd /home/aistudio/PaddleGAN/applications/!mkdir output!export PYTHONPATH=$PYTHONPATH:/home/aistudio/PaddleGAN && python -u tools/first-order-demo.py  --driving_video ~/work/京剧.MP4  --source_image ~/work/京剧.png --relative --adapt_scale
print("生成的视频路径/home/aistudio/PaddleGAN/applications/output/result.mp4")

Python# 库安装!pip install moviepy

Python#为生成的视频加上音乐 保存的路径为 ---->  /home/aistudio/PaddleGAN/applications/output/京剧result.mp4from moviepy.editor import *
# videoclip_1放入想要加入的音频视频文件后续会提取音频。videoclip_1 = VideoFileClip("/home/aistudio/work/京剧.MP4")# videoclip_2 放入上述生成的无声视频。videoclip_2 = VideoFileClip("./output/result.mp4")
audio_1 = videoclip_1.audio
videoclip_3 = videoclip_2.set_audio(audio_1)
videoclip_3.write_videofile("./output/京剧result.mp4", audio_codec="aac")

print("添加音频后的视频路径为/home/aistudio/PaddleGAN/applications/output/京剧result.mp4")