一种目标定位模型的训练方法和目标定位方法及装置
基本信息
申请号 | CN201810992851.7 | 申请日 | - |
公开(公告)号 | CN109165654B | 公开(公告)日 | 2021-03-30 |
申请公布号 | CN109165654B | 申请公布日 | 2021-03-30 |
分类号 | G06K9/20(2006.01)I;G06K9/34(2006.01)I;G06K9/62(2006.01)I | 分类 | 计算;推算;计数; |
发明人 | 叶锦宇;刘玉明 | 申请(专利权)人 | 北京九狐时代智能科技有限公司 |
代理机构 | 北京超凡志成知识产权代理事务所(普通合伙) | 代理人 | 苏胜 |
地址 | 100000北京市朝阳区阜通东大街1号院5号楼2单元324002 | ||
法律状态 | - |
摘要
摘要 | 本申请提供了一种目标定位模型的训练方法,包括:将图像样本输入到卷积模型,以提取图像样本的第一图像特征;将所述第一图像特征输入到分割模型,以生成图像样本的第一前景坐标;将所述第一图像特征输入到回归模型,以生成图像样本的第二前景坐标;根据所述第一前景坐标和第二前景坐标计算模型损失函数;根据模型损失函数对卷积模型、分割模型和回归模型同时进行训练,以生成由卷积模型和回归模型组成的目标定位模型。采用上述方案,通过训练卷积模型、分割模型和回归模型,使得实际预测阶段的卷积模型获得更好的图像特征,同时,利用回归模型提升图像定位速度。 |
