Image enhancement method and device fusing wavelet transform and attention mechanism

The invention provides an image enhancement method and device fusing wavelet transform and an attention mechanism, and the method comprises the steps: inputting a target image into a feature extraction branch model in an enhancement model, and obtaining a plurality of first feature maps of different scales of the target image; inputting the target image into a first down-sampling module of an image enhancement branch model in the enhancement model to obtain a second feature map of the target image; inputting the second feature map and the first feature map into a first up-sampling module of an image enhancement branch model to obtain a third feature map of the target image; the first up-sampling module and the first down-sampling module are both fused with wavelet transform and an attention mechanism; and inputting the third feature map of the target image into a third output layer of the image enhancement branch model to obtain an enhanced image of the target image. According to the method, the wavelet trans