An image processing method, device and storage medium based on reinforcement learning

Abstract

The present application discloses an image processing method, device and storage medium based on reinforcement learning, which relate to the technical field of image processing and are used to solve the problem of poor processing effect due to inability to accurately process images. In this method, feature extraction is performed on the target image to obtain a feature map of the target image; the feature map is input into the first network based on reinforcement learning to obtain the target processing method of each pixel in the target image; wherein, the first network is It is obtained by training the target network model based on the reinforcement learning method; in the target network model, mask scoring is performed on each spatial position of the feature map extracted by at least one specified neural network layer; the target image is processed according to the target processing method. In this way, by combining reinforcement learning and spatial attention mechanism, it can help the neural network to understand and extract image features, and then ensure that the obtained strategy is more reasonable, thereby improving the effect of image processing.

Description

technical field [0001] The present application relates to the technical field of image processing, and in particular, to an image processing method, device and storage medium based on reinforcement learning. Background technique [0002] With the popularization of smartphones and the continuous upgrading of camera equipment, users can easily obtain photos to record the wonderful moments in life. However, in order to meet the characteristics of convenient portability, the current mobile phones inevitably make some compromises on the camera hardware, which leads to a big difference in the image quality captured by smartphones compared with advanced devices such as digital SLR cameras. The most common hardware limitation is the lack of a small sensor composed of a large aperture lens and a photodiode, resulting in insufficient light input, insufficient color, high noise, and poor contrast compared with the images captured by advanced devices such as SLRs. . At present, most o...

AI Technical Summary

Problems solved by technology

[0005] Embodiments of the present disclosure provide an image processing method and related devices to solve the problem in the prior art that the image cannot be processed accurately, resulting in poor processing effect

Images