网上有很多使用PS给黑白老照片上色的教程,下图是P上颜色的爱因斯坦:
于是,有大牛使用深度学习制作了一个自动给黑白图像上色的模型,非常非常的牛叉。
- colornet
- Automatic Colorization
- Hypercolumns for Object Segmentation and Fine-grained Localization
- The major advancements in Deep Learning in 2016
本帖就使用前文的妹子图训练这个模型,看看效果。
由于原作者使用的TensorFlow版本太旧,不能直接运行,我做了一点修改。
训练模型:
import tensorflow as tf # 0.12
import numpy as np
import os
import glob
import sys
from matplotlib import pyplot as plt
# 训练文件列表
filena