一种基于YOLOv3改进模型的用于检测PCB中圆形目标的方法
摘要:
本发明公开了一种基于YOLOv3改进模型的用于检测PCB中圆形目标的方法,首先对获取的印制电路板的X射线图像进行标注和数据集划分;然后利用设定的圆形anchor替换YOLOv3模型中的矩形anchor,使YOLOv3改进模型可以直接预测圆形目标的边界,接着对损失函数进行修改;其次,对所述YOLOv3改进模型进行迭代优化和验证,并基于非极大值抑制算法对三个检测层所有预测的圆形目标中低置信度的以及重叠较大的目标进行过滤,输出检测结果,相比于传统的圆形目标检测方法,本发明的精确性与鲁棒性更好;改进后的模型与原模型相比参数更少,速度更快且误检率更低。
0/0