基于Tiny-YOLOv3改进算法的工件识别

Workpiece recognition based on improved Tiny-YOLOv3 algorithm

  • 摘要: 针对Tiny-YOLOv3算法在工件识别实时检测中存在漏检率高的问题, 提出了在Tiny-YOLOv3基础上加以改进实现了对工件更加快速、准确地识别。主要改进的方式是在Tiny-YOLOv3的特征提取网络中增加3个网络模块,即SPP结构、SE模块和Ghost模块, 并用卷积层代替池化层, 改进后的网络结构平均精度均值、准确率和网络模型大小都有着显著的改善。试验结果表明,改进后的算法能够更好的提升工件识别的效率,并同时满足在嵌入式设备中进行实时检测的要求。

     

    Abstract: For the Tiny-YOLOv3 algorithm, the problem of high missed detection rate in real-time detection of workpiece recognition, Based on the improvement of Tiny-Yolov3, this paper realizes faster and more accurate recognition of small workpieces. The Tiny-Yolov3 feature extraction network adds three network modules, namely the SPP structure, the SE module and the Ghost module, and uses a convolutional layer to replace the pooling layer. Therefore, the average accuracy, accuracy and size of the network model of the improved network structure have been significantly improved. Experimental results show that the improved algorithm proposed in this paper can improve the efficiency of workpiece recognition and meet the requirements of real-time detection of embedded devices.

     

/

返回文章
返回