摘要
本发明提供一种基于改进YOLOv5的带钢表面缺陷检测方法及系统,其中方法包括:搭建改进的anchor‑freeYOLOv5网络;获取带钢表面图像;将所述带钢表面图像输入至所述anchor‑freeYOLOv5网络,获得带钢表面缺陷检测结果;输出所述带钢表面缺陷检测结果。本发明的基于改进YOLOv5的带钢表面缺陷检测方法及系统,降低超参数的数量提高检测速度,针对数据不平衡、训练速度慢的问题对损失函数进行改进,采用EIoU loss作为回归损失函数,其尺度不变,可以直接优化边框回归的评价指标加快模型拟合速度,为防止大量负样本造成网络的退化,采用焦距损失作为客观性损失,Focal Loss通过增加损失权重,改善了难分类样本的检测精度。