Improved YOLOV3 target recognition algorithm for adaptive edge optimization
摘要
为了准确识别出图像中的目标零件
本文提出一种自适应边缘优化的改进YOLOV3目标识别算法.首先
利用K-means++算法计算出适用于本文数据集的anchor box
接着设计自适应边缘误差函数
并与改进PSO算法结合
得到改进YOLOV3算法(YOLOV3-AEEF).然后采集零件图像并进行数据增强
标注图片
得到样本集.加载预训练权重后对网络进行训练
并在测试集上测试.实验结果表明
当样本图片中存在较多残缺零件干扰时
YOLOV3存在将背景识别为零件的情况
而YOLOV3-AEEF能够准确识别出目标零件
在保证较高查全率的情况下
查准率较YOLOV3提高21%
提升了网络的综合性能.
Abstract
In order to accurately identify the target part in the image
an adaptive edge optimization YOLOV3 target recognition algorithm was proposed. Firstly
the K-means++ algorithm is used to calculate the anchor box suitable for the data set of this paper. Then the adaptive edge error function is designed and combined with the improved PSO algorithm to obtain the YOLOV3-AEEF algorithm. Then collect the part image and enhance the data
label the picture
and get the sample set. The network is trained after loading the pre-training weights and tested on the test set. The experimental results show that when there are more incomplete parts interference in the sample picture
YOLOV3 identifies the background as a part
and YOLOV3-AEEF can accurately identify the target part
and in the case of a higher recall
the precision is 21% higher than YOLOV3
which improves the overall performance of the network.