Image colorization based on transformer

期刊论文
出版模式：

开放获取
发表日期：

2025年7月1日

查看全文

Min Xu

摘要

This paper presents a transformer-based method for colorizing grayscale image. By employing a deep architecture with stacked encoder-decoder layers, the model effectively captures intricate features, significantly improving its expressive capacity. The encoder primarily extracts features from the input grayscale image, while the decoder utilizes these features alongside the self-attention mechanism and contextual information to predict the corresponding \(a^*\) and \(b^*\) chrominance components in CIELAB color space, and combines the channel \(L^*\) of the input image to achieve image colorization. To ensure the generated image mimics a real-life image in terms of pixel accuracy and visual quality, the algorithm employs a variety of loss functions. Experimental results demonstrate significant performance improvements in black and white image colorization. The generated images exhibit natural coloring and rich details, making them highly valuable in practical applications.