Paper
14 March 2022 Weather image classification based on generative adversarial network and transfer learning
Yonglong Zou, Jiaxin Wu, Weizhe Chen, Yang Liu
Author Affiliations +
Abstract
In view of the low recognition accuracy of traditional weather recognition methods and the serious imbalance in the number of weather images in various categories in the weather image data set, a weather image classification algorithm based on generative adversarial network and transfer learning is proposed to solve the above problems. The proposed method mainly includes two parts: data equalization based on generative adversarial network and image classification based on transfer learning. This paper uses generative adversarial network to amplify the data of a few categories of weather images, so as to obtain a relatively balanced weather image data set.The method of transfer learning is used to fine-tune the model to realize the classification of weather images. The experimental results show that the method proposed in this paper is better than the traditional method, effectively solving the problem of low model classification accuracy caused by the imbalance of training samples, and realizing the recognition and classification of four types of weather images: sunny, foggy, rainy, and snowy.
© (2022) COPYRIGHT Society of Photo-Optical Instrumentation Engineers (SPIE). Downloading of the abstract is permitted for personal use only.
Yonglong Zou, Jiaxin Wu, Weizhe Chen, and Yang Liu "Weather image classification based on generative adversarial network and transfer learning", Proc. SPIE 12165, International Conference on Intelligent Traffic Systems and Smart City (ITSSC 2021), 121651S (14 March 2022); https://doi.org/10.1117/12.2627879
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Image classification

Data modeling

Data conversion

Image processing

Performance modeling

Neural networks

Statistical modeling

Back to Top