项目场景:
利用飞桨实现该应用过程中出现问题:AnimeGAN动漫化模型一键应用(含动漫化小程序体验)https://aistudio.baidu.com/aistudio/projectdetail/1308514?channelType=0&channel=0
原始代码:
#由于PaddleHub升级比较快,建议大家直接升级到最新版本的PaddleHub
!pip install paddlehub -U -i https://pypi.tuna.tsinghua.edu.cn/simple
# 该模型需要gpu跑,所以需要设置下gpu环境
%env CUDA_VISIBLE_DEVICES=0
%matplotlib inline
import matplotlib.pyplot as plt
import matplotlib.image as mpimg
# 待输入图片
test_img_path = ["./test.jpg"]
# 展示图片
img1 = mpimg.imread(test_img_path[0])
plt.figure(figsize=(10,10))
plt.imshow(img1)
plt.axis('off')
plt.show()
# 转换为宫崎骏《起风了》风格图片
import cv2
import paddlehub as hub
# 模型加载
# use_gpu:是否使用GPU进行预测
model = hub.Module('animegan_v1_h