The invention discloses a
wavelet attention shadow removal method based on soft prior guidance, and belongs to the field of
image processing. The method comprises the following steps: constructing a multi-scale
wavelet attention codec architecture, decomposing an input image into a low-frequency component representing illumination characteristics and a high-frequency component containing texture details by using
discrete wavelet transform, and establishing multi-scale
frequency domain representation; proposing probabilistic soft prior modeling, adaptively generating a continuous differentiable probability
distribution diagram by using a convolutional sub-network and a double-slope
Sigmoid function, and guiding differentiated repair of high and
low frequency features in a dynamic weight mode; and cross-domain illumination correction based on gray world
hypothesis is introduced,
spatial domain enhancement is performed on a
frequency domain restoration result, illumination consistency of cross-domain
processing is ensured, and finally a shadow-free image is output. According to the method,
wavelet domain feature optimization is guided through the probabilistic soft
mask, high-quality shadow removal is realized, and the robustness and visual quality of shadow removal are remarkably improved while the light weight of the model is kept.