刘学平, 李玙乾, 刘励, 王哲. 自适应边缘优化的改进YOLOV3目标识别算法[J]. 微电子学与计算机, 2019, 36(7): 59-64.
引用本文: 刘学平, 李玙乾, 刘励, 王哲. 自适应边缘优化的改进YOLOV3目标识别算法[J]. 微电子学与计算机, 2019, 36(7): 59-64.
LIU Xue-ping, LI Yu-qian, LIU Li, WANG Zhe. Improved YOLOV3 target recognition algorithm for adaptive edge optimization[J]. Microelectronics & Computer, 2019, 36(7): 59-64.
Citation: LIU Xue-ping, LI Yu-qian, LIU Li, WANG Zhe. Improved YOLOV3 target recognition algorithm for adaptive edge optimization[J]. Microelectronics & Computer, 2019, 36(7): 59-64.

自适应边缘优化的改进YOLOV3目标识别算法

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.

     

/

返回文章
返回