使用paddlehub着色黑白图片

背景

家里有很多小时候的老照片，都是交卷洗出来的黑白照片，看到网上有app可以把照片着色，于是想自己动手处理下。

使用工具paddlehub

PaddleHub是基于PaddlePaddle生态下的预训练模型管理和迁移学习工具，可以结合预训练模型更便捷地开展迁移学习工作。便捷地获取PaddlePaddle生态下的所有预训练模型，涵盖了图像分类、目标检测、词法分析、语义模型、情感分析、语言模型、视频分类、图像生成、图像分割等主流模型。

github地址：paddlehub

主要是自然语言和计算机视觉算法。

paddlehub安装

1 2	!pip install --upgrade paddlepaddle -i https://mirror.baidu.com/pypi/simple !pip install --upgrade paddlehub -i https://mirror.baidu.com/pypi/simple

paddlehub着色图片

import paddlehub as hub
from PIL import Image
import cv2

model = hub.Module(name='deoldify') #加载模型比较慢，需要等会
input_img = 'aaaa.jpeg' #输入目录下的图片
result = model.predict(input_img) #图片上色，非常的吃cpu，我的mac基本卡死
cv2.imwrite('bbbb.jpg', result[0]) #输出图片到文件

最终效果

对比图