この論文では、Capsule Network(CapsNet)の単純なトポロジーを画像の色付けの問題について調査します。元のCapsNetトポロジの生成およびセグメンテーション機能は、画像分類問題用に提案されており、次のようにネットワークを変更することで画像の色付けに活用されます:1)元のCapsNetモデルは、グレースケール入力を出力にマッピングするように適合されていますCIE Labカラースペースで、2)モデルの特徴検出器部分は、低レベルの画像表現機能をこのモデルに転送するために、VGG-19事前トレーニングモデルから継承されたより深い特徴層を使用して更新されます。3)マージン損失関数は、平均二乗誤差(MSE)損失として変更され、イメージ間のマッピングが最小化されます。結果のCapsNetモデルはColorizer Capsule Network(ColorCapsNet)という名前になります。ColorCapsNetのパフォーマンスはDIV2Kデータセットで評価され、画像の色付けの問題についてCapsule Networksをさらに調査するための有望な結果が得られます。
In this paper, a simple topology of Capsule Network (CapsNet) is investigated for the problem of image colorization. The generative and segmentation capabilities of the original CapsNet topology, which is proposed for image classification problem, is leveraged for the colorization of the images by modifying the network as follows:1) The original CapsNet model is adapted to map the grayscale input to the output in the CIE Lab colorspace, 2) The feature detector part of the model is updated by using deeper feature layers inherited from VGG-19 pre-trained model with weights in order to transfer low-level image representation capability to this model, 3) The margin loss function is modified as Mean Squared Error (MSE) loss to minimize the image-to-imagemapping. The resulting CapsNet model is named as Colorizer Capsule Network (ColorCapsNet).The performance of the ColorCapsNet is evaluated on the DIV2K dataset and promising results are obtained to investigate Capsule Networks further for image colorization problem.