摘要
本发明涉及图像识别技术领域,具体是一种基于最优传输距离损失模型的表面杂质检测方法,包括以下操作步骤:S1、获取玻璃表面图像数据,并保存在数据库中;S2、在基础Yolov5网络结构的输出层中增加用于检测微小杂质的微小杂质特征图,以构成改进Yolov5网络结构,并优化定位损失函数;S3、对获取的玻璃表面图像数据进行切割缩小处理,将处理后的数据输入到改进Yolov5网络结构中进行训练,并对训练完成的玻璃杂质检测模型进行优化,以得到优化后的玻璃杂质检测模型;S4、将新的玻璃表面图像输入到优化后的玻璃杂质检测模型中进行检测,以得到各个污渍的类别、位置和大小;本发明通过增设的微小杂质特征图,能够有效的提高网络结构预测的精准度。
- 单位