今天继续给大家分享有趣的 AI 项目。
今天我们分享用NoGAN
的图像增强技术给老照片着色。效果如下:
原图
上色后
NoGAN
是一种新型GAN
,它能花费最少的时间进行GAN
训练。
今天分享的这个项目已在GitHub
开源项目,下面我们来运行一下。
1. 准备工作
首先,用git clone
命令下载源码
git clone https://github.com/jantic/DeOldify.git
进入项目根目录,安装Python
依赖包
pip3 install -r requirements.txt
编写代码运行项目之前,需要下载预训练好的模型。项目提供了三个模型
模型
区别如下:
-
ColorizeArtistic_gen.pth: