首页/ 重庆创通联达智能技术有限公司/ 专利详情

训练缺陷分级检测模型的方法、装置、设备及存储介质

基本信息

申请号 CN202110420016.8 申请日 -
公开（公告）号 CN113205176A 公开（公告）日 2021-08-03
申请公布号 CN113205176A 申请公布日 2021-08-03
分类号 G06N3/04;G06N3/08;G06K9/62 分类 计算；推算；计数;
发明人 高航;杜松 申请（专利权）人 重庆创通联达智能技术有限公司
代理机构 北京东方亿思知识产权代理有限责任公司 代理人 彭琼
地址 401120 重庆市渝北区仙桃数据谷中路99号4幢
法律状态 -

摘要

摘要 本发明实施例提供了一种训练缺陷分级检测模型的方法、装置、设备及存储介质。该方法包括：获取样本图像；将样本图像输入预设的缺陷分级检测模型；利用缺陷分级检测模型中的分类分支网络和分级分支网络对样本图像进行缺陷分级检测，得到样本图像中缺陷的类别、位置和级别的预测值；利用缺陷分级检测模型中的聚类分支网络对样本图像进行处理得到样本图像中缺陷级别的伪标签，并将为标签作为样本图像中缺陷级别的真值；根据缺陷类别、位置、级别的预测值和真值训练预设的缺陷分级检测模型，得到目标缺陷分级检测模型。根据本发明实施例可以获得能够进行缺陷级别检测的模型，且训练过程中无需进行缺陷级别的标注。