公开(公告)号：CN118365567A

公开(公告)日：2024-07-19

申请号：CN202410610025.7

申请日：2024-05-16

Applicant: 北京理工大学

Inventor： 艾丹妮 ,  杨浩宇 ,  杨健 ,  范敬凡 ,  宋红 ,  付天宇 ,  肖德强 ,  林毓聪

IPC: G06T5/77 ,  G06T7/90 ,  G06T7/44 ,  G06N3/045 ,  G06N3/0464 ,  G06N3/0475 ,  G06N3/094

Abstract: 本发明公开了一种图像颜色修复方法、装置、设备及存储介质，该方法使用CycleGAN做为主要网络框架。实现了滤镜图像和无滤镜图像之间的转换。为了提高生成图像的颜色真实性以及图像纹理恢复，对生成器进行了改进。通过增加位置编码注意力显式地编码输入特征图的坐标信息，生成的图像颜色更加真实纹理更加清晰。利用通道和空间注意力机制对生成网络进行了优化，并通过多尺度判别器对图像在不同尺度进行细致的分析和判别，以达到精确校正滤光片引起的颜色失真，从而提升医学图像特征匹配和位姿估计任务的性能。