物体缺陷识别模型的训练方法、物体缺陷识别方法及装置

基本信息

申请号 CN202110023478.6 申请日 -
公开(公告)号 CN112766110A 公开(公告)日 2021-05-07
申请公布号 CN112766110A 申请公布日 2021-05-07
分类号 G06N3/08(2006.01)I;G06K9/62(2006.01)I;G06N3/04(2006.01)I;G06K9/00(2006.01)I;G06K9/32(2006.01)I 分类 计算;推算;计数;
发明人 蔡蓉;李廷 申请(专利权)人 重庆创通联智物联网有限公司
代理机构 北京市隆安律师事务所 代理人 权鲜枝;何健
地址 401336重庆市南岸区玉马路8号科技创业中心融英楼(经开区拓展区内)
法律状态 -

摘要

摘要 本申请公开了一种物体缺陷识别模型的训练方法、物体缺陷识别方法及装置,该物体缺陷识别模型的训练方法包括:获取物体的训练样本图像;通过物体缺陷识别模型中的缺陷定位分割子模型确定有缺陷训练图像中的缺陷区域,并将其从有缺陷训练图像中分割出来,得到缺陷区域图像;通过物体缺陷识别模型中的缺陷分类子模型对缺陷区域图像进行缺陷类别的识别,得到缺陷类别识别结果和缺陷类别识别损失值;根据缺陷类别识别损失值更新物体缺陷识别模型的参数。本申请的训练方法能够避免图像中其他无关信息对于缺陷类别识别的干扰,使得模型的识别效率和准确率均较高,且训练得到的模型对于不同类型的缺陷的泛化能力较好,降低了物体缺陷识别的成本。