The invention discloses an
image restoration method based on spectrum guidance enhancement. The
image restoration method comprises a
diffusion model; the
diffusion model comprises a forward
noise adding process and a reverse denoising process; in the reverse denoising process, a U-Net neural network is adopted to carry out denoising restoration on the
noise-added image to obtain a restored image; in the U-Net neural network, spectral
decomposition is carried out on low-order features of an
encoder through a spectrum guidance module to obtain high-frequency components and low-frequency components,
weight distribution is carried out on the high-frequency components and the high-order features through a trans-attention mechanism, the high-frequency components and the high-order features are spliced with up-sampling features of a decoder, and local high-frequency features of an image are restored; fusing the low-frequency component with the up-sampling feature of the decoder, and repairing the global low-frequency feature of the image; training and testing the
diffusion model introduced into the spectrum guidance module by adopting a
training set and a
test set divided by the image
data set; inputting an image needing to be repaired into the diffusion model introduced into the
frequency spectrum guidance module to obtain a high-resolution repaired image; and the restored image is clearer and more accurate.