Classification of Multi-Class Weather Image Data
In order to improve the accuracy of multi-class weather image recognition, a weather image classification algorithm based on ViT was proposed. The multi-layer transformer method is used for feature extraction and encoder. The self attention method is introduced to calculate the similarity between ea...
Saved in:
Published in | 2023 9th International Symposium on System Security, Safety, and Reliability (ISSSR) pp. 203 - 207 |
---|---|
Main Authors | , , , , |
Format | Conference Proceeding |
Language | English |
Published |
IEEE
01.06.2023
|
Subjects | |
Online Access | Get full text |
Cover
Loading…
Summary: | In order to improve the accuracy of multi-class weather image recognition, a weather image classification algorithm based on ViT was proposed. The multi-layer transformer method is used for feature extraction and encoder. The self attention method is introduced to calculate the similarity between each position in the segmented image dataset. Then, all positions are weighted according to the score to generate a new feature representation. Finally, the weather image data is classified by MLP classifier. Experiments show that the deep learning algorithm based on ViT can effectively improve the prediction accuracy of the model, and has a good effect on improving the recognition of multi-class weather images. The algorithm can achieve good accuracy on 14 different weather data, with a recognition accuracy of up to 92.83%. |
---|---|
ISSN: | 2835-2823 |
DOI: | 10.1109/ISSSR58837.2023.00038 |