集成注意力机制基于YOLOv5开发构建布匹瑕疵检测识别系统

商品的瑕疵、缺陷类型的检测本身就是AI与生产制造也相互结合的爆点,在之前的文章中也有做过这块的项目实践,但是整体的效果并不是很好,主要也跟自己的数据集有很大的关系,这里简单给出前文的链接,如有需要可自行移步阅读即可。

《基于YOLO实践布匹缺陷检测》

本文的核心目的就是基于公开源数据集开发构建不同类型的YOLOv5检测识别模型,来挖掘在布匹瑕疵检测领域内模型的上限,首先看下效果图:

简单看下数据集:

 这里一共开发了三款模型,分别是s系列、m系列已经基于s系列融合CBAM注意力的提升模型,在我前文都有对应的yaml文件,这里就不再赘述了,默认都是100次epoch的迭代计算,结果详细如下:

 Batch计算实例:

 接下来综合对比分析一下三款模型的性能:

F1值曲线:

 精确率曲线:

 召回率曲线:

 loss曲线:

 从评估结果上直观对比来看:融合CBAM注意力后s系列的模型性能有所提升,但整体还是不如m系列的模型,这个本身m系列的模型参数里也是s的几倍,自然是没法比的。

猜你喜欢

转载自blog.csdn.net/Together_CZ/article/details/130282111