发明名称 |
语义分割模型训练和图像分割方法及装置、计算设备 |
摘要 |
本发明公开了一种语义分割模型训练和图像分割方法及装置、计算设备,属于计算机视觉技术领域,其中方法包括:将训练图像输入至语义分割模型,得到语义分割模型输出的训练图像的语义分割的初步结果;依据弱监督信息和从训练图像中选择的多个局部候选区域,进行局部候选区域融合,得到训练图像的语义分割的校正结果;依据初步结果和校正结果,对语义分割模型的模型参数进行修正;迭代执行训练步骤直至所述语义分割模型的训练结果满足预定收敛条件。本发明方案既有像素级别的直接监督,又可以端到端的优化语义分割模型,又能够根据对局部候选区域的判断改善分割分支的结果。 |
申请公布号 |
CN106530305A |
申请公布日期 |
2017.03.22 |
申请号 |
CN201610848794.6 |
申请日期 |
2016.09.23 |
申请人 |
北京市商汤科技开发有限公司 |
发明人 |
石建萍 |
分类号 |
G06T7/11(2017.01)I;G06T7/194(2017.01)I;G06T7/136(2017.01)I |
主分类号 |
G06T7/11(2017.01)I |
代理机构 |
北京市浩天知识产权代理事务所(普通合伙) 11276 |
代理人 |
宋菲;刘云贵 |
主权项 |
一种语义分割模型训练方法,其特征在于,所述方法采用多个预先标注有弱监督信息的训练图像对语义分割模型进行训练,针对于每一个训练图像,训练步骤包括:将所述训练图像输入至所述语义分割模型,得到所述语义分割模型输出的所述训练图像的语义分割的初步结果;依据所述弱监督信息和从所述训练图像选择的多个局部候选区域,进行局部候选区域融合,得到所述训练图像的语义分割的校正结果;依据所述初步结果和所述校正结果,对所述语义分割模型的模型参数进行修正;迭代执行所述训练步骤直至所述语义分割模型的训练结果满足预定收敛条件。 |
地址 |
100084 北京市海淀区中关村东路1号院3号楼7层710-712房间 |