[发明专利]一种基于强化学习的图像处理方法、装置及存储介质

摘要

本申请公开了一种基于强化学习的图像处理方法、装置及存储介质，涉及图像处理技术领域，用以解决因无法准确的处理图像从而导致处理效果差的问题。该方法中，对目标图像进行特征提取，得到目标图像的特征图；将特征图输入到基于强化学习的第一网络中，得到目标图像中各像素点的目标处理方式；其中，第一网络是基于强化学习的方法训练目标网络模型得到的；在目标网络模型中，对至少一个指定神经网络层提取的特征图的各空间位置进行掩码打分处理；根据目标处理方式对目标图像进行处理。这样，通过将强化学习和空间注意力机制结合，能够很好的帮助神经网络理解和提取图像特征，进而保证得出的策略是更加合理的，从而提高图像处理的效果。