The invention discloses a multi-
label weather image
data expansion method based on DiT condition guidance, and mainly solves the problems that an existing enhancement method is single, data is scarce in multi-
label weather classification, and the control capability of a generation model is insufficient. According to the scheme, the method comprises the steps of 1) obtaining a training sample set containing multi-
label weather images; 2) constructing a
diffusion model based on LDM and DiT condition guidance and training the
diffusion model; 3) converting the image into
potential space representation through a VAE
encoder, and gradually adding
Gaussian noise; de-noising is carried out by using the trained
diffusion model in combination with multi-label condition embedding, and a synthetic weather image is recovered through a VAE decoder; and 4) mixing the synthesized weather image with original
training set data to obtain an expanded
data set for optimization training of a multi-label classification model. The method can effectively improve the diversity and quality of a
data set, remarkably improves the precision and robustness of multi-label weather classification, and can be used for a multi-label weather recognition task in
deep learning.