一种水下图像轻量化处理方法和装置

本发明涉及水下图像处理领域，公开了一种水下图像轻量化处理方法和装置，其技术方案要点是，包括如下步骤：选定教师模型和学生模型，获取历史图像数据，并划分数据集，使用数据集对选定的教师模型进行训练得到固化的教师模型；将数据集分别输入到教师模型和选定的学生模型，分别输出得到特征图；对学生模型输出特征图进行筛选，输出筛选后的掩膜特征图；将掩膜特征图与教师模型输出特征图归一化后结合求取损失；将学生模型输出特征图和教师模型特征图解耦，求取预测损失；使用数据集训练学生模型，求取损失；将、、通过各自的权重结合相加得到最终损失，并轮次迭代进行蒸馏学习，得到成品学生模型；将成品学生模型部署应用。