一种基于异构双分类头的医学图像分割方法

本发明涉及人工智能的医学图像半监督分割领域，具体涉及一种基于异构双分类头的医学图像分割方法，包括获取包括有标签数据集和无标签数据集的医学图像数据集；构建初始模型，其包括主干网络、线性分类器和原型分类器；采用有标签数据集对初始模型进行全监督训练得到预热模型；采用Two‑Stream Batch Sample方式对医学图像数据集进行处理，得到多个批量；采用一个批量对预热模型进行自适应伪标签置信度训练，得到线性预测结果集和原型预测结果集；基于线性预测结果集和原型预测结果集，计算训练损失并反向传播优化预热模型；重复上述操作直至模型参数收敛，得到医学图像分割模型；采用医学图像分割模型对待处理图像进行分割。