基于注意力机制的条件生成对抗网络在恶劣环境下的图像处理方法、装置、设备、存储介质和程序产品
申请号:CN202510621489
申请日期:2025-05-14
公开号:CN120543428A
公开日期:2025-08-26
类型:发明专利
摘要
本公开涉及图像处理技术领域,尤其是提供一种基于注意力机制的条件生成对抗网络在恶劣环境下的图像处理方法、装置、设备、存储介质和程序产品。方法包括:获取待处理的带雾图像;利用预先训练的包含生成器的图像去雾模型,根据带雾图像生成去雾图像,图像去雾模型的训练样本包括带雾训练图像和相应的去雾训练图像;生成器包括:编码器,用于编码带雾图像,生成第一特征图;注意力模块,用于根据第一特征图的各个通道间的动态关系及非线性关系,得到归一化的通道权重,并将通道权重与第一特征图相乘得到第二特征图;解码器,用于根据第二特征图,生成去雾图像。实施本公开的技术方案,可以生成更真实的去雾图像,降低颜色失真和减少信息损失。
技术关键词
图像去雾模型
去雾图像
条件生成对抗网络
混合损失函数
图像处理方法
解码器
编码器
注意力机制
通道
全局平均池化
非线性
图像处理装置
图像获取模块
压缩特征
图像处理技术
关系
计算机程序产品