ISTFormer: lightweight transformer for enhanced super-resolution of coal rock images via iterative feature extraction
摘要
Coal rock image acquisition often suffers from poor quality due to challenges such as inadequate lighting, noise, and sensor limitations. To address these issues, we introduce ISTFormer, a lightweight transformer-based approach for coal rock image super-resolution. ISTFormer integrates a global permuted self-attention block with a local convolutional block, leveraging iterative up-and-down sampling to extract comprehensive features from low-resolution images. Experiments on benchmark datasets and a custom coal rock dataset, CD-188, demonstrate ISTFormer’s superiority over existing lightweight super-resolution methods, achieving state-of-the-art results with improved PSNR and SSIM scores. By open-sourcing our algorithm’s code and datasets, we aim to foster further developments in this field. This work presents a significant step toward enhancing coal rock image quality for improved identification and classification accuracy. Our source code is available at: https://github.com/haoliuliuhao/ISTFormer.