一种面向无序摆放的焊条头视觉计数方法

本发明公开了一种面向无序摆放的焊条头视觉计数方法，包括：S1：采集焊条头的图像，得到真实数据集；对真实数据集进行扩充，最终得到训练集和测试集；S2：基于YOLOv8架构，采用DS‑PD‑CONV卷积模块和NON_LOCAL注意力机制对YOLOv8的网络结构进行改进，得到焊条头检测网络模型，本发明提供了一种面向无序摆放的焊条头视觉计数方法，通过对YOLOv8算法进行深度优化改进，提高焊条头关键特征点的提取能力，然后调整特征识别时网络的注意力机制，并提出两步检测策略减少焊条头数量过多时的漏检测问题，最终实现高于99.5％正确率的焊条头计数精度，本发明的方法具有较强的稳定性、扩展性，可应用于常见类型的多目标计数、细小目标识别等情况。