An efficient coding‐based grayscale image automatic colorization method combined with attention mechanism

doi:10.1049/ipr2.12452

An efficient coding‐based grayscale image automatic colorization method combined with attention mechanism

2022-02-22

0

OA

AI

DOI：10.1049/ipr2.12452

原文链接

查看原文

摘要

En 中文

AbstractThe development of deep learning provides a new way for solving the colorization problem on the grayscale image. Excellent coding‐based methods appear in the automatic image colorization task, avoiding the unsaturated colour effect problem of previous methods based on the L2 loss function. Traditional neural networks come with high computational costs and a large number of parameters. Considering the limitation of memory and computing resources and aiming at lightweight, a novel grey image automatic colorization network is proposed. The basic idea of coding‐based methods is used, regarding the colorization task as a pixel‐level classification problem, meanwhile redesign and improve the colour encoding and decoding process. This network architecture leverages a lightweight convolution to reduce the computation and combines an efficient attention model to form a residual block as the kernel of the backbone network. Furthermore, an efficient image self‐attention mechanism placed at the end of the network is applied to enhance the ultimate colouring results. The method proposed in this paper can maintain the natural colouring effect and significantly reduce the computational amount and network model parameters.

An efficient coding‐based grayscale image automatic colorization method combined with attention mechanism

摘要

AI总结

期刊

机构

引用论文