公开(公告)号：CN119205532A

公开(公告)日：2024-12-27

申请号：CN202411250178.1

申请日：2024-09-06

Applicant: 南京理工大学

Inventor： 万敏杰 ,  佘梦云 ,  顾国华 ,  陈钱 ,  贺晴 ,  钱惟贤 ,  王佳节

IPC: G06T5/60 ,  G06T5/90 ,  G06N3/0455 ,  G06N3/0464 ,  G06N3/084 ,  G06N3/0985

Abstract: 本发明公开了一种基于消色差先验引导的无监督颜色恒常方法，涉及图像处理领域，包括：设计U型编解码颜色恒常网络模型，将网络公开大型无光源标签数据集作为训练集，输入网络模型进行训练；设计空间注意力机制，加入模型进一步提升性能；设计无监督训练方法，训练过程不依赖场景中光源实际颜色的知识，训练网络检测彩色图像转换成灰度或者梯度后的消色差像素，并根据损失函数梯度调整网络参数，直到最大迭代次数，输出网络模型；将带有真实光源标签的测试集图像输入训练好的网络模型，输出颜色校正的图像。经本发明改进后的网络模型在图像光源估计与真值的误差、图像的视觉效果和设备的兼容性得到了明显提升。