使用基于End2end CNN的模型进行照片着色!
彩色照片可以为您的日常生活带来很多乐趣。 试想一下,如果您给奶奶的旧照片上色并将其展示给家人,他们会感到多么惊讶和高兴。
使用Photoshop为图片着色是一个非常累人的过程,因此我们实现了此“图片着色”工具,该工具基于“ ,它是基于CNN的深度学习,它可以为任何具有任何色阶和图片分辨率的灰度图片着色。
:dizzy: 版本0.1出来了!
:open_book: 文献资料
怎么跑
安装要求:
pip3 install -r requirements.txt
在本地运行Web应用
python3 WebApp/app.py
然后端点将在上运行
如何使用预训练模型
从此处下载预训练的模型: :
将文件解压缩到algorithm/Models/
再次运行“运行端点”
结果
结果一点也不差! 很多测试用例会如此逼真,但是模型仍然需要更多的训练
原始图像
2021-10-11 09:03:40
91.38MB
Python
1