公开(公告)号：CN116630175A

公开(公告)日：2023-08-22

申请号：CN202310270044.5

申请日：2023-03-20

Applicant: 新疆大学

Inventor： 陈娟 ,  郭磊 ,  周艳云 ,  石飞 ,  马海钰 ,  肖怒 ,  郭济 ,  王志杰 ,  陈震 ,  王隆太

IPC: G06T5/00 ,  G06N3/08 ,  G06V10/44 ,  G06V10/82

Abstract: 本发明公开了一种基于GNN‑transformer架构的单幅图像去雪方法，属于计算机图像处理技术领域，首先将需去雪图像输入到GNN‑transformer框架中，使用对比学习训练出的残差网络实现基于图像的图构建；其次，使用训练好的图卷积网络模型对带雪图进行特征提取，并引入注意力机制来定位雪花位置，聚合特征信息；最后将提取到雪花特征生成雪花注意力图，将其输入到生成对抗网络中进行去雪操作，最终输出去雪图像。本发明示例的一种基于GNN‑transformer架构的单幅图像去雪方法，能够高效地通过图像中的雪花结构和纹理特征来识别雪的位置，并将其去除；学习图像细节特征的同时准确完成图像去雪任务，恢复被雪覆盖的图像，从而提高图像质量。