LAN: Lightweight Attention-based Network for RAW-to-RGB Smartphone Image Processing
Daniel Wirzberger Raimundo, Andrey Ignatov, Radu Timofte
Abstract
The number of pictures taken by smartphones is growing exponentially. However, the smartphones’ limitations both in size and cost negatively impact on the quality of the implemented sensors. At the same time, their computing power has also been steadily improving, allowing the usage of more complex processing methods to enhance images. In prior works, deep neural networks trained with matched sensor outputs and DSLR images have shown to bring substantial improvements to the images, compared to classical and handcrafted methods. We propose a lightweight attention-based network (LAN) that employs a convolutional layer to learn the input mosaic and an unsupervised pre-training strategy. Our method is validated on standard benchmarks and shown to improve over the state-of-the-art in both perceptual and fidelity terms without hindering GPU inference time on smartphone devices. Our code is available at: github.com/draimundo/LAN