The invention relates to the technical field of
deep learning, and particularly relates to an
infrared rainfall prediction and model training method and device, and
electronic equipment, and the method comprises the steps: obtaining a
training set, each sample in the
training set comprising a multi-channel
brightness temperature feature of a sample region and a corresponding rainfall feature true value; a preset
infrared rainfall prediction model is obtained, the
infrared rainfall prediction model comprises an input layer, an
encoder, a middle layer, a decoder and an output layer, the
encoder is additionally provided with an attention mechanism module before down-sampling each time, the middle layer comprises a cavity space
pyramid pooling module, the decoder is additionally provided with an attention mechanism module after up-sampling each time, and the output layer comprises a cavity space
pyramid pooling module; a weighted
mean square error
loss function of the model dynamically adjusts an error weight according to
precipitation spatial heterogeneity; and training a preset infrared rainfall prediction model by using the
training set to obtain a trained infrared rainfall prediction model. According to the technical scheme, infrared
rainfall estimation can be carried out more accurately.