针对工业视觉应用的预训练数据筛选及训练方法、系统

基本信息

申请号 CN202110753850.9 申请日 -
公开(公告)号 CN113420824A 公开(公告)日 2021-09-21
申请公布号 CN113420824A 申请公布日 2021-09-21
分类号 G06K9/62(2006.01)I;G06N3/04(2006.01)I;G06N3/08(2006.01)I 分类 计算;推算;计数;
发明人 严鼎天;张亮;孙海;高全亮;郑荣;周涛;丁香 申请(专利权)人 上海理想信息产业(集团)有限公司
代理机构 上海方澜知识产权代理事务所(普通合伙) 代理人 李娜
地址 201315上海市浦东新区秀沿西路189号电信信息园区B4楼
法律状态 -

摘要

摘要 本发明实施例涉及工业场景视觉检测技术应用领域,公开了一种多针对工业视觉应用的预训练数据筛选及训练方法、系统。该方法包括:利用开源数据集训练目标检测网络得到预训练网络M1;利用预训练网络M1计算并统计得到开源数据集中每个类别的特征,建立特征库L1;利用检测目标的训练集训练调整预训练网络M1;利用调整后的预训练网络M1统计得到检测目标的训练集中每个类别的特征;利用距离函数从特征库L1中筛选出与检测目标的每个类别的特征匹配的目标类别;保存开源数据集中与目标类别对应的训练数据作为目标检测网络的目标预训练数据。本发明实施例可从开源数据集中挖掘与当前任务相关性更强的预训练数据,从而提升模型的准确度,稳定性和可用性。