图像分割模型训练方法、装置、计算机设备和存储介质

基本信息

申请号 CN202110390733.0 申请日 -
公开(公告)号 CN113112509A 公开(公告)日 2021-07-13
申请公布号 CN113112509A 申请公布日 2021-07-13
分类号 G06T7/11(2017.01)I 分类 计算;推算;计数;
发明人 赖昕;田倬韬;刘枢;贾佳亚;沈小勇;吕江波 申请(专利权)人 深圳思谋信息科技有限公司
代理机构 广州华进联合专利商标代理有限公司 代理人 纪婷婧
地址 518051广东省深圳市南山区高新南九道45号三航科技大厦22楼
法律状态 -

摘要

摘要 本申请涉及一种图像分割模型训练方法、装置、计算机设备和存储介质。所述方法包括:获取第一样本图像、第二样本图像,以及与第一样本图像对应的标注图像;将第一样本图像和标注图像以及第二样本图像输入至待训练的图像分割模型,以使待训练的图像分割模型根据第一样本图像对应的预测图像以及标注图像,得到第一损失值,以及根据第一图像块和第二图像块的特征图关于图像重叠区域的特征相似度,得到第二损失值;第一图像块和第二图像块为第二样本图像中具有图像重叠区域的图像块;基于第一损失值和第二损失值,训练得到图像分割模型。采用本方法能够提高得到的图像分割模型的图像分割准确率。