基于残差Swin Transformer的天气图像识别技术研究  

Research on Weather Image Recognition based on Residual Swin Transformer

作　　者：张卓然 张倩 宋智 何嘉[1] ZHANG Zhuoran;ZHANG Qian;SONG Zhi;HE Jia(College of Computer Science,Chengdu University of Information Technology,Chengdu 610225,China;Active Network(Chengdu),Ltd.,Chengdu 610000 China;Sichuan Meteorological Detection Data Center,Chengdu 610072,China)

机构地区：[1]成都信息工程大学计算机学院,四川成都610225 [2]活跃网络(成都)有限公司,四川成都610000 [3]四川省气象探测数据中心,四川成都610072

出　　处：《成都信息工程大学学报》2023年第6期637-642,共6页Journal of Chengdu University of Information Technology

基　　金：四川省科技厅资助项目(2021005);四川省重点实验室科技发展基金资助项目(2018-青年-11)。

摘　　要：人类活动经常受到天气条件的影响,基于图像的自动天气识别在实际应用中具有重要意义。然而现有方法均使用卷积神经网络,未能有效地利用图像的全局信息和像素点之间长距离的依赖关系,且识别的天气类型较少,识别精度较低。为解决这些问题,尝试将视觉Transformer应用到天气识别领域,同时提出一种基于残差Swin Transformer的模型,并使用先进的优化器Ranger来提高天气识别的正确率。该模型在包含11种天气现象的公开数据集WEAPD上进行验证,实验结果表明,其整体性能优于其他先进的识别网络,识别正确率达到93.6%,可为天气图像识别和天气预报研究提供参考。Human activities are often affected by weather conditions,and automatic weather recognition-based image is of great importance in practical applications.However,existing methods all use convolutional neural networks,which fail to effectively utilize the global information of images and the long-distance dependency between pixel points,and recognize fewer weather types with low recognition accuracy.To solve these problems,we try to apply the visual Transformer to the field of weather recognition,and also propose a model based on the residual Swin Transformer and use the advanced optimizer Ranger to improve the weather recognition rate.The model is validated on WEAPD,a publicly available dataset containing 11 weather phenomena,and the results show that its overall performance is better than other advanced recognition networks,with a 93.6%correct recognition rate.It can benefit the research of weather image recognition and weather forecasting.

关 键 词：天气现象 图像识别 深度学习 Swin Transformer 

分 类 号：TP391.4[自动化与计算机技术—计算机应用技术]