【技术实现步骤摘要】
本专利技术涉及雪天图像处理领域，尤其涉及一种雪天图像恢复方法。
技术介绍
1、雪天图像恢复是计算机视觉中图像恢复的主要类型之一，根据不同的雪花大小可以分为小雪花、中雪花、大雪花。
2、现有的去雪方法基本分为三种类型。第一种是传统方法，如使用滤波和颜色空间变换等易于实现的方法。第二种方法是采用端到端的深度学习去雪框架。第三种是针对所有天气恢复的框架。
3、针对不同类型的雪天图像，现有的算法都不能完全清除，特别是大雪花图像，现有方法恢复会产生模糊和伪影，极难恢复。具体而言在自动驾驶领域，汽车在雪天行驶，飘落的雪花会对自动驾驶系统分析周围环境造成遮挡，这对于行驶安全造成很大隐患，很容易导致交通事故。同时在目标检测领域，有雪图像会影响行人检测、人员跌倒、行人越界等一系列的高级任务。由于雪花的无规则特征，如雪花大小和形状的不固定性，雪晶生成过程中吸附不同的微小颗粒，造成透明度的问题，雪花飘落轨迹受大气影响造成雪花飘落轨迹不可预测。另外传统的除雪方法依赖于先验知识，所以人为因素对除雪效果有很大影响。传统方法一般是单独检索雪花位置，但由于雪天复杂多变，所以会存在很多类型雪花没有检测到，生成的清晰图像可能会产生变形或颜色失调等问题，这同时造成了有雪图像恢复成为一项更具挑战性的问题。
技术实现思路
1、为了解决现有方法在去雪过程中存在的颜色偏差，遮蔽效应，伪影等技术问题，本专利技术提出了一种雪天图像恢复方法、设备及存储介质，方法是采用一种在雪天气条件下端到端的融合快速傅里叶和
2、具体地，方法包括以下步骤：
3、s1、构建多尺度雪天图像恢复网络；所述多尺度雪天图像恢复网络包括：基于快速傅里叶变换金字塔模型的编码器模块、特征融合模块、基于globalvit的解码器模块；
4、s2、训练所述雪天图像恢复网络，得到训练完成的网络；
5、s3、利用训练完成的网络进行雪天图像恢复。
6、一种存储介质，所述存储介质存储指令及数据用于实现所述的一种雪天图像恢复方法。
7、一种雪天图像恢复设备，包括：处理器及存储介质；所述处理器加载并执行存储介质中的指令及数据用于实现所述的一种雪天图像恢复方法。
8、本专利技术提供的有益效果是：设计了一种在雪天气条件下端到端的融合快速傅立叶和全局视觉transformer(globalvit)的框架ftnet，这是一种有效且新颖的基于transformer的图像去雪架构，它是由卷积神经网络块的编码器和globalvit的解码器组成，成功解决了去雪过程中颜色偏差，遮蔽效应，伪影等问题。
本文档来自技高网...【技术保护点】
1.一种雪天图像恢复方法，其特征在于：包括以下步骤：
2.如权利要求1所述的一种雪天图像恢复方法，其特征在于：所述基于快速傅里叶变换金字塔模型的编码器模块包括：傅里叶变换单元、卷积单元、反傅里叶变换单元、金字塔模块(PPM)单元和Relu单元。
3.如权利要求1所述的一种雪天图像恢复方法，其特征在于：所述基于GlobalViT的解码器模块包括：LinearQ、K、V单元和多轴门控MAG单元。
4.如权利要求3所述的一种雪天图像恢复方法，其特征在于：所述LinearQ单元用雪查询模块代替，雪查询模块采用隐藏层。
5.如权利要求1所述的一种雪天图像恢复方法，其特征在于：步骤S2中多尺度雪天图像恢复网络处理图像的过程如下：
6.如权利要求2所述的一种雪天图像恢复方法，其特征在于：所述基于快速傅里叶变换金字塔模型的编码器模块的内部处理过程如下：
7.如权利要求4所述的一种雪天图像恢复方法，其特征在于：所述基于GlobalViT的解码器模块的内部处理过程如下：
8.如权利要求3所述的一种雪天图像恢复方法，其
9.一种存储介质，其特征在于：所述存储介质存储指令及数据用于实现权利要求1～8任一项所述的一种雪天图像恢复方法。
10.一种雪天图像恢复设备，其特征在于：包括：处理器及存储介质；所述处理器加载并执行存储介质中的指令及数据用于实现权利要求1～8任一项所述的一种雪天图像恢复方法。
...【技术特征摘要】
1.一种雪天图像恢复方法，其特征在于：包括以下步骤：
2.如权利要求1所述的一种雪天图像恢复方法，其特征在于：所述基于快速傅里叶变换金字塔模型的编码器模块包括：傅里叶变换单元、卷积单元、反傅里叶变换单元、金字塔模块(ppm)单元和relu单元。
3.如权利要求1所述的一种雪天图像恢复方法，其特征在于：所述基于globalvit的解码器模块包括：linearq、k、v单元和多轴门控mag单元。
4.如权利要求3所述的一种雪天图像恢复方法，其特征在于：所述linearq单元用雪查询模块代替，雪查询模块采用隐藏层。
5.如权利要求1所述的一种雪天图像恢复方法，其特征在于：步骤s2中多尺度雪天图像恢复网络处理图像的过程如下：
...【专利技术属性】
技术研发人员：魏佳璇，陈炜龙，谢天，杨毅，刘晋海，徐小听，
申请(专利权)人：兰州大学，
类型：发明
国别省市：
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。