辽宁石油化工大学学报
  期刊
  出版年
  关键词
结果中检索 Open Search
Please wait a minute...
选择: 显示/隐藏图片
基于图像分块和多头注意力机制的气象识别研究
赵旭峰, 刘琳琳, 曹宇, 叶成荫, 郭宗凯
辽宁石油化工大学学报    2024, 44 (2): 83-90.   DOI: 10.12422/j.issn.1672-6952.2024.02.013
摘要1203)   HTML16)    PDF (1793KB)(122)    收藏

基于图像对天气现象进行识别,对天气状况的分析至关重要。针对传统的机器学习方法对各类天气特征难以准确提取且天气现象分类效果差,以及深度学习对天气现象识别的准确率不高的问题,提出了基于图像分块和多头注意力机制的天气识别模型。该模型首次将Swin Transformer引入天气识别领域中,采用了窗口多头自注意层与移位窗口多头自注意层相结合的多头注意力机制。结果表明,其区域相关特征提取能力弥补传统方法的不足,能够提取图像中复杂的天气特征。采用迁移学习对模型进行训练,将微调模型的全连接参数输入到Softmax分类器,实现了对多类别天气图像的识别,识别准确率为99.20%,优于对比的几种主流方法。因此,该方法可以作为天气识别模块应用于地面气象识别系统。

图表 | 参考文献 | 相关文章 | 多维度评价