摘要
本申请涉及语义分割技术领域,具体涉及一种基于加权对比网络半监督语义分割和模型训练方法及装置。训练方法包括:获取含有目标物体的有标签图像集和对应的标签集,以及含有目标物体且无标签的无标签图像集;将有标签图像集分别输入至第一分支的第一基础网络和第二分支的第二基础网络,对应得到第一目标掩码集和第二目标掩码集;将无标签图像集输入第一基础网络、第二基础网络、第一增强网络和第二增强网络,对应得到第三目标掩码集、第四目标掩码集、第一增强伪标签集和第二增强伪标签集;基于总差异度调整第一基础网络、第二基础网络的参数,以完成语义分割模型的训练。改善了现有技术由于样本类别不均衡导致性能不佳的问题。
技术关键词
标签
网络
语义分割方法
基础
像素点
物体
分支
模型训练方法
语义分割模型训练
语义分割技术
参数更新模块
电子设备
图像获取模块
训练系统
计算机