一种基于残差注意力网络和对比学习的红外图像真彩转换方法及系统

    公开(公告)号:CN118097241A

    公开(公告)日:2024-05-28

    申请号:CN202410162530.X

    申请日:2024-02-05

    Abstract: 本发明公开了一种基于残差注意力网络和对比学习的红外图像真彩转换方法及系统,包括:将红外成像设备采集到的红外图像输入到红外图像真彩转换网络,利用基于空间和通道注意力的残差注意力模块捕捉输入的红外图像的空间特征,以及将不同特征之间进行混合和整合;利用通道注意力模块使信息量较大的通道得到更多的重视,利用空间注意力模块通过聚类每个位置的特征以增强网络的表示能力;通过生成器根据输入的红外图像重建出对应的可见光图像;利用多尺度判别器将生成的可见光图像分成若干小块,并针对每个小块进行真实性评估,并在通过评估后进行输出。本发明达到降低模型的复杂性并提高收敛性、扩大研究场景以及实现高分辨红外图像真彩转换的目的。

Patent Agency Ranking