オブジェクトの色収差を排除する新しい方法として、計算による色の恒常性は、多くのコンピュータビジョンアプリケーションの基本的な前提条件になっています。このタスクを実行するアルゴリズムの中で、学習ベースのアルゴリズムは近年大きな成功を収めています。しかし、画像の空間情報を十分に考慮しておらず、光源推定の精度を向上させる余地が十分にあります。本論文では、画像の空間情報を活用することにより、畳み込みニューラルネットワーク(CNN)を使用したAttention Dense Color Constancy(ADCC)と呼ばれる色恒常性アルゴリズムを提案します。具体的には、入力画像の2Dログクロミナンスヒストグラムとそれらの特別に拡張されたヒストグラムに基づいて、ADCCは自己注意DenseNetで光源を推定します。増強された画像は、対数ヒストグラムのエッジグラデーション、エッジピクセル、および非エッジピクセルを区別するのに役立ちます。これらは、特徴抽出と色のあいまいさの排除に大きく貢献し、それによって光源推定の精度を向上させます。ベンチマークデータセットのシミュレーションと実験は、提案されたアルゴリズムが最先端の方法と比較して光源の推定に効果的であることを示しています。したがって、ADCCは、色がオブジェクトを区別するための重要な要素であるスマートカメラなどのスマートシティのアプリケーションを促進する大きな可能性を提供します。
As a novel method eliminating chromatic aberration on objects, computational color constancy has becoming a fundamental prerequisite for many computer vision applications. Among algorithms performing this task, the learning-based ones have achieved great success in recent years. However, they fail to fully consider the spatial information of images, leaving plenty of room for improvement of the accuracy of illuminant estimation. In this paper, by exploiting the spatial information of images, we propose a color constancy algorithm called Attention Dense Color Constancy (ADCC) using convolutional neural network (CNN). Specifically, based on the 2D log-chrominance histograms of the input images as well as their specially augmented ones, ADCC estimates the illuminant with a self-attention DenseNet. The augmented images help to tell apart the edge gradients, edge pixels and non-edge ones in log-histogram, which contribute significantly to the feature extraction and color-ambiguity elimination, thereby advancing the accuracy of illuminant estimation. Simulations and experiments on benchmark datasets demonstrate that the proposed algorithm is effective for illuminant estimation compared to the state-of-the-art methods. Thus, ADCC offers great potential in promoting applications of smart cities, such as smart camera, where color is an important factor for distinguishing objects.