摘要
本发明公开了车载夜视系统中基于注意力机制的图像细节恢复方法,包括:获取低光照夜视图像的图像张量;将所述图像张量输入至预训练的图像生成模型,对其执行特征映射;根据所述特征映射的映射参数,在输出层生成细节恢复图像;其中,所述特征映射的映射参数表征为图像生成模型在预训练中学习到的SRGB标准图像的色彩矩阵和伽马值;本发明通过将卷积层提取的局部特征与自注意力层捕捉的全局依赖特征相结合,弥补了现有图像增强方法的不足;在处理低光照图像时,卷积层专注于细节的恢复,而自注意力层则能捕捉到图像中远距离像素之间的关系,从而有效地恢复全局信息,使得恢复图像在细节与全局结构上都得到增强。
技术关键词
车载夜视系统
图像生成模型
恢复方法
注意力机制
对比度
网络架构
更新模型参数
亮度
色彩
矩阵
依赖特征
生成参数
像素点
样本
信号处理
定义
双通道架构