Guiding the Illumination Estimation Using the Attention Mechanism (CROSBI ID 689307)
Prilog sa skupa u zborniku | izvorni znanstveni rad | međunarodna recenzija
Podaci o odgovornosti
Koščević, Karlo ; Subašić Marko ; Lončarić, Sven
engleski
Guiding the Illumination Estimation Using the Attention Mechanism
Deep learning methods have achieved a large step forward in many computer vision applications. With mechanisms such as attention, deep models can now guide themselves to focus on parts of an image that are more significant for a given task. In computational color constancy, the most important step is to estimate the illumination vector as accurately as possible. Since illumination estimation algorithms can be sensitive to noise, such as ambiguous regions in the image, the ability to have a mechanism to look for specific regions in an image could be helpful. In this paper, a convolutional neural network with an attention mechanism is proposed. The attention mechanism helps the network to focus on regions that contain more content and to avoid regions where ambiguous estimations may occur. In the experimental results, it is shown that the attention mechanism does help the network to obtain more accurate estimations and puts the focus of the network on the regions in an image where gradients are high. The network with the attention mechanism achieves up to 10% increase in accuracy compared to the same network architecture without the attention mechanism.
neural network ; image processing ; deep learning ; convolution ; computational color constancy ; regression ; attention mechanism ; white balancing
nije evidentirano
nije evidentirano
nije evidentirano
nije evidentirano
nije evidentirano
nije evidentirano
Podaci o prilogu
143-149.
2020.
objavljeno
10.1145/3379310.3379329
Podaci o matičnoj publikaciji
Proceedings of the 2020 2nd Asia Pacific Information Technology Conference
Association for Computing Machinery (ACM)
9781450376853
Podaci o skupu
2nd Asia Pacific Information Technology Conference (APIT 2020)
predavanje
17.01.2020-19.01.2020
Bali, Indonezija