|国家科技期刊平台
首页|期刊导航|红外技术|基于轻量化多尺度下采样网络的红外图像非均匀性校正算法

基于轻量化多尺度下采样网络的红外图像非均匀性校正算法OA北大核心CSTPCD

Infrared Image Non-uniformity Correction Algorithm Based on Lightweight Multiscale Downsampling Network

中文摘要英文摘要

红外成像系统常由于探测单元的非均匀性导致成像结果出现条纹噪声.基于深度学习的红外图像非均匀校正算法为取得较好的校正结果,通常采用复杂度高的网络结构,导致计算量庞大.本文提出了一种轻量化网络的红外图像非均匀校正算法,并针对 Unet 网络的编码过程设计了一种轻量化多尺度下采样模块(Lightweight Multi-scale Downsampling Module,LMDM).LMDM通过像素拆分和通道重构实现特征图下采样,利用多个串联的深度可分离卷积(Depth-wise Separable Convolution,DSC)实现多尺度特征提取.此外,该算法引入轻量化通道注意力机制用于调整特征权重,实现更好的上下文信息融合.实验结果表明,与对比算法相比,本文提出的算法在保证校正图像纹理清晰、细节丰富和边缘锐利的前提下,内存占用降低 70%以上,红外图像处理速度提升 24%以上.

Infrared imaging systems often produce fringe noise in imaging results owing to the non-uniformity of the detection unit.To obtain better correction results,most deep learning-based infrared image non-uniformity correction algorithms adopt complex network structures,which increase the computational cost.This study proposes a lightweight network-based infrared image non-uniformity correction algorithm and designs a lightweight multi-scale downsampling module(LMDM)for the encoding process of the Unet network.The LMDM uses pixel splitting and channel reconstruction to realize feature map downsampling and realizes multi-scale feature extraction using multiple cascaded depth-wise separable convolutions(DSC).In addition,the algorithm introduces a lightweight channel attention mechanism for adjusting feature weights to achieve better contextual information fusion.The experimental results show that the proposed algorithm reduces memory use by more than 70%and improves the processing speed of the infrared images by more than 24%compared with the comparison algorithm while ensuring that the corrected image has a clear texture,rich details,and sharp edges.

牟新刚;朱太龙;周晓

武汉理工大学 机电工程学院,湖北 武汉 430070

计算机与自动化

红外图像非均匀性校正深度学习轻量化多尺度特征提取

infrared imagenon-uniformity correctiondeep learninglightweightmulti-scale feature extraction

《红外技术》 2024 (005)

501-509 / 9

国家自然科学基金项目(61701357);中央高校基本科研业务费专项资金资助(183204007).

评论