白黒画像の全自動色付け
本研究では、ディープネットワークを用いて白黒画像をカラー画像に自動変換する手法を提案する。提案手法では、画像の大域特徴と局所特徴を考慮した新たな畳込みネットワークモデルを用いることで、画像全体の構造を考慮した自然な色付けを行うことができる。提案モデルにおいて、大域特徴は画像全体から抽出され、局所特徴はより小さな画像領域から計算される。これらの特徴は“結合レイヤ”によって一つに統合され、色付けネットワークに入力される。このモデル構造は入力画像のサイズが固定されず、どんなサイズの画像でも入力として用いることができる。また、モデルの学習のために既存の大規模な画像分類のデータセットを利用し、それぞれの画像の色とラベルを同時に学習に用いることで、効果的に大域特徴を学習できるようにしている。提案手法により、100年前の白黒写真など、様々な画像において自然な色付けを実現できる。色付けの結果はユーザテストによって評価し、約90%の色付け結果が自然であるという回答が得られた。
論文
2016年
Let there be Color!: Joint End-to-end Learning of Global and Local
Image Priors for Automatic Image Colorization with Simultaneous
Classification
Satoshi Iizuka*, Edgar Simo-Serra*, Hiroshi Ishikawa (* equal contribution)
ACM Transactions on Graphics (SIGGRAPH), 2016
We present a novel technique to automatically colorize grayscale images that
combines both global priors and local image features. Based on Convolutional
Neural Networks, our deep network features a fusion layer that allows us to
elegantly merge local information dependent on small image patches with global
priors computed using the entire image. The entire framework, including the
global and local priors as well as the colorization model, is trained in an
end-to-end fashion. Furthermore, our architecture can process images of any
resolution, unlike most existing approaches based on CNN. We leverage an
existing large-scale scene classification database to train our model,
exploiting the class labels of the dataset to more efficiently and
discriminatively learn the global priors. We validate our approach with a user
study and compare against the state of the art, where we show significant
improvements. Furthermore, we demonstrate our method extensively on many
different types of images, including black-and-white photography from over a
hundred years ago, and show realistic colorizations.
@Article{IizukaSIGGRAPH2016, author = {Satoshi Iizuka and Edgar Simo-Serra and Hiroshi Ishikawa}, title = {{Let there be Color!: Joint End-to-end Learning of Global and Local Image Priors for Automatic Image Colorization with Simultaneous Classification}}, journal = "ACM Transactions on Graphics (SIGGRAPH)", year = 2016, volume = 35, number = 4, }
ソフトウェア
This code is the implementation of the "Let there be Color!: Joint End-to-end
Learning of Global and Local Image Priors for Automatic Image Colorization with
Simultaneous Classification" paper. It contains the pre-trained model and example
usage code.