点击下方卡片,关注“OpenCV与AI深度学习”公众号!
导读本文给大家分享一个用OpenCV DNN模块给黑白老照片上色的实例,并给出Python和C++版本源码。
背景介绍这个项目是基于在加利福尼亚大学,伯克利,Richard Zhang,Phillip Isola和Alexei A、Efros开发的研究工作--Colorful Image Colorization,对应论文地址:https://arxiv.org/pdf/1603.08511.pdf,作者项目github地址:https://github.com/richzhang/colorization/tree/caffe
正如在最初的论文中所解释的,作者们接受了问题的潜在不确定性,将其作为一项分类任务,在训练时使用类别再平衡来增加结果中的颜色多样性。人工智能(AI)方法在测试时在CNN(“卷积神经网络”)中作为前馈传递实现,并在100多万张彩色图像上进行训练。
这个项目将使用的颜色空间模型是“Lab”。CIELAB颜色空间(也称为CIE L*a*b*或有时简称为“Lab”颜色空间)是国际照明委员会(CIE)在1976年定义的颜色空间。它将颜色表示为三个数值,L*表示亮度,a*和b*表示绿色、红色和蓝黄色。
深度学习的过程: