RGB and LUT based Cross Attention Network for Image Enhancement
Tengfei Shi (Beihang University), Chenglizhao Chen (China University of Petroleum (East China)),* Yuanbo He (State Key Laboratory of Virtual Reality Technology and Systems, Beihang University), wenfeng song (Beijing Information Science and Technology University), Aimin Hao (BeihangUniversity)
The 34th British Machine Vision Conference
Abstract
Image enhancement aims to improve the quality of images by adjusting their color and is widely used in professional digital photography. Deep learning-based 3 Dimensional LookUp-Table (3D LUT) of RGB color transformation has achieved promising performance in terms of speed and precision. However, the focus has mainly been on building an adaptive enhancer by only learning the global color adjusting weights from the image, which ignores the significant relationship between the intrinsic semantic information of the image and LUT that is relevant to photographers. In this paper, we propose the Cross Attention Network (CANet), a new framework that formulates image enhancement as a parallel learning process based on the image and LUT features. To better learn the adjustment weights for both global color and intrinsic semantics, we propose a cross attention architecture that connects low-level (color, edge and outline) and high-level (semantic) features of the image and color transform LUT features to generate appropriate adjustment weights. Meanwhile, we employ a LUT-Aware Module (LAM) to construct the channels and spatial attention for refining the LUT features. Since these modules have a more powerful representational capacity, they can better capture the intrinsic relationship between image semantics and LUT features. The extensive evaluations on standard benchmarks, including FiveK and HDR datasets, show that CANet achieves better performance compared to state-of-the-art methods.Citation
@inproceedings{Shi_2023_BMVC,
author = {Tengfei Shi and Chenglizhao Chen and Yuanbo He and wenfeng song and Aimin Hao},
title = {RGB and LUT based Cross Attention Network for Image Enhancement},
booktitle = {34th British Machine Vision Conference 2023, {BMVC} 2023, Aberdeen, UK, November 20-24, 2023},
publisher = {BMVA},
year = {2023},
url = {https://bmvc2022.mpi-inf.mpg.de/BMVC2023/0348.pdf}
}
Copyright © 2023
The British Machine Vision Conference is organised by