基于最优阈值的曲线结构图像分割半监督训练方法及装置

本发明公开了基于最优阈值的曲线结构图像分割半监督训练方法及装置，涉及计算机视觉领域，基于教师模型对未标注数据进行预测，得到伪标签数据；在预设的阈值搜索空间内，以预设步长进行搜索，直至找到一个使伪标签分布与预设标记先验分布之间KL散度最小的阈值，将其确定为最优阈值；基于最优阈值对伪标签数据进行二值化处理，得到二值化后的伪标签数据；使用二值化后的伪标签数据和已标注数据的真实标签计算总损失函数，使用总损失函数更新学生模型参数；基于当前步参数通过滑动平均方法更新教师模型参数，教师模型参数由学生模型参数的指数移动平均获得。本发明训练方法提高了在已标注数据有限而未标注数据充沛情况下的曲线结构图像分割性能。