用深度學習給黑白照片著色

Deep Learning 近幾年在圖像識別的進步有目共睹,UC Berkeley 的一個團隊剛發布的研究嘗試用 Deep Learning 給黑白照片著色。題圖就是用該模型做出的對比圖,左側是一張民國老照片,南京諾那塔。

給黑白照片著色是從灰度一維到 RGB 三維的轉換,這是一個無定解的(underconstrained)問題。這篇文章的大致思路是通過幾百萬張彩色照片進行訓練,對物件(包括背景色)進行分類、提取特徵,從而嘗試給出對灰度圖片的色彩預期。簡單說就是嘗試給深度學習演算法以常識,告訴它天是藍的,草是綠的,蒼老師是世界的 ^_^

完整的文章、訓練好的模型可以從論文作者的網站上找到:Colorful Image Colorization

搭建好環境後(講真,Caffe 裝起來不是一點半點的折騰)測試了幾張照片,效果時好時壞,但偶爾能看到驚艷的:

福利時間:如果這篇文章贊多的話 (騙贊小能手) 我可以幫轉幾張黑白照片 :-)

推薦閱讀:

TAG:深度學習DeepLearning | Caffe深度學習框架 | 計算機視覺 |