一种交互式小样本语义分割的训练方法

基本信息

申请号 CN202111223958.3 申请日 -
公开(公告)号 CN113936178A 公开(公告)日 2022-01-14
申请公布号 CN113936178A 申请公布日 2022-01-14
分类号 G06V10/764(2022.01)I;G06V10/774(2022.01)I;G06V10/778(2022.01)I;G06K9/62(2022.01)I 分类 计算;推算;计数;
发明人 杨培文;张成英;梁惠莹;于振东;张辽 申请(专利权)人 深圳市玻尔智造科技有限公司
代理机构 杭州九洲专利事务所有限公司 代理人 张羽振
地址 518110广东省深圳市龙华区观澜街道大富社区大富工业区20号硅谷动力智能终端产业园A5栋101
法律状态 -

摘要

摘要 本发明涉及一种交互式小样本语义分割的训练方法,包括步骤:在原始图片上绘制不同类别的点,用点标注待学习目标;将标注出待学习目标的图片输入预训练模型,进行特征提取;将标注点投影到特征图上得到特征向量和类别标签,使用逻辑回归模型对原始标注点进行分类和预测;将单张图片作为小样本,学习小样本不同类别标注点的特征。本发明的有益效果是:通过对样本图片进行交互式的标注学习目标、特征提取与训练、过检和漏检修复等操作,可以达到仅使用一张图也可以训练出效果不错的模型效果,解决了当前目标检测常见方法中标注数据工作量大和使用小样本数据训练模型泛化能力弱的问题。