利用深度学习实现灰度图像的自动彩色化。
autocolorize的Python项目详细描述
自动着色
使用机器学习自动着色图像。
安装
确保您有caffe(使用python绑定)。然后运行:
pip install autocolorize
运行:
autocolorize grayscale.png -o colorized.png
api
您还可以从python中进行着色(假设grayscale是要着色的图像):
import autocolorize classifier = autocolorize.load_default_classifier() rgb = autocolorize.colorize(grayscale, classifier=classifier)