一种基于扩散模型的水下图像增强方法、装置及介质

本发明涉及一种基于扩散模型的水下图像增强方法、装置及介质，其中方法包括以下步骤：获取退化图像；将退化图像分别输入教师模型和学生模型；计算教师模型和学生模型的轨迹；输入退化图像至反射估计模块，计算反射图；通过扩散模型的正向扩散过程，将反射图映射为教师模型的轨迹修正量，并以预设的权重将轨迹修正量和教师模型的轨迹逐位相加，修正教师模型的轨迹；计算修正后的教师模型轨迹和学生模型轨迹的蒸馏损失，冻结教师模型的参数，基于梯度下降优化学生模型参数；利用优化后的学生模型对退化图像进行图像增强，得到增强后的水下图像。与现有技术相比，本发明具有在降低计算量的同时提升了模型在细节和纹理回复方面的性能等优点。