一种基于EfficientNet的螺母表面缺陷分类方法
基本信息
| 申请号 | CN202110289071.8 | 申请日 | - |
| 公开（公告）号 | CN113052809A | 公开（公告）日 | 2021-12-10 |
| 申请公布号 | CN113052809A | 申请公布日 | 2021-12-10 |
| 分类号 | G06T7/00;G06T3/60;G06T7/11;G06T7/136;G06T7/194;G06T7/187;G06K9/62 | 分类 | 计算；推算；计数; |
| 发明人 | 李子杰;程坦;刘涛;吕剑 | 申请（专利权）人 | 中科海拓（无锡）科技有限公司 |
| 代理机构 | 安徽善安知识产权代理事务所（特殊普通合伙） | 代理人 | 陈庭 |
| 地址 | 214000 江苏省无锡市滨湖区新城置业大厦20楼 | ||
| 法律状态 | - | ||
摘要
| 摘要 | 本发明公开了一种基于EfficientNet的螺母表面缺陷分类方法，包括步骤一：采集训练样本，制作数据集；通过数据集训练EfficientNet模型，模型训练收敛后，再进行螺母表面分类；通过使用EfficientNet对模型规模进行自调整，避免人工调整费时，且难以得到最优解；步骤二：采集目标图像，对目标图像进行裁剪；步骤三：修改损失函数为focal loss，调整正负样本权重，提升模型精度；免了正负样本不均衡对模型的影响；步骤四：图像预处理，制作模板图像对输入图像进行旋转矫正，并做差得到差值图像，将三者融合为三通道图像作为模型输入图像；增强模型对生产环境的适应能力，提升模型的泛化能力，步骤五：模型推理，使用EfficientNet模型对输入图像进行推理；步骤六：推理图像输出到上位机显示。 |
