Journal of Liaoning Petrochemical University
  Journals
  Publication Years
  Keywords
Search within results Open Search
Please wait a minute...
For Selected: Toggle Thumbnails
Research on Weather Recognition Based on Image Segmentation and Multi⁃Head Attention Mechanism
Xufeng ZHAO, Linlin LIU, Yu CAO, Chengyin YE, Zongkai GUO
Abstract1203)   HTML16)    PDF (1793KB)(122)      

Recognition of weather phenomena based on images is essential for the analysis of weather conditions. To address the problems that traditional machine learning methods are difficult to accurately extract various weather features and poor in classifying weather phenomena and the accuracy of deep learning for weather phenomena recognition is not high, a weather recognition model based on image block and multi?headed attention mechanism is proposed. The model introduces Swin Transformer into the field of weather recognition for the first time, and adopts a multi?headed attention mechanism combining window multi?head self?attention layer and shifted?window multi?head self?attention layer, whose regionally relevant features extraction capability makes up for the shortcomings of traditional methods and can extract complex weather features from images. The model is trained using transfer learning, and the fully connected parameters of the fine?tuned model are input to the Softmax classifier to achieve recognition of multi?category weather images with 99.20% recognition accuracy, which is better than several mainstream methods in comparison, and it can be applied to ground weather recognition systems as a weather recognition module.

2024, 44 (2): 83-90. DOI: 10.12422/j.issn.1672-6952.2024.02.013