基于双向特征金字塔的YOLOv4的改进

IMPROVEMENT OF YOLOV4 BASED ON BIDIRECTIONAL FEATURE PYRAMID

  • 摘要: 为了进一步提高YOLOv4算法的目标检测精度,提出一种改进算法YOLOv4TB(YOLOv4+Transformer+BiFPN)。JP+1该算法利用Transformer提取同一特征图上的共同发生的对象特征来进行特征增强,利用BiFPN(Bi-directional Feature Pyramid Network)模型替换PAN模型,解决YOLOv4中存在的冗余计算和不同特征层贡献度相同的问题。并在此基础上采用Leaky-ReLU激活函数和可分离卷积技术,解决目标检测精度下降和参数量、运算量上升的问题。在PASCAL VOC数据集上实验结果表明,与YOLOv4相比,YOLOv4TB算法具有较高的检测精度和运行效率,参数数量和运算量有所减少。

     

    Abstract: In order to further improve the target detection accuracy of YOLOv4 algorithm, an improved YOLOv4TB (YOLOV4+Transformer+BIFPN) algorithm is proposed. The proposed algorithm used Transformer to extract co-occurring object features on the same feature map for feature enhancement, and replaced the PAN model with BIFPN (bi-directional feature pyramid network) model. The problems of redundant calculation and the same contribution of different feature layers in YOLOv4 were solved. On this basis, the Leaky-ReLU activation function and the separable convolution technology were used to solve the problems of the decreased accuracy of target detection and the increase of the number of parameters and computation. Experimental results on Pascal VOC data sets show that compared with YOLOv4, YOLOv4TB algorithm has higher detection accuracy and operating efficiency, and the number of parameters and computation are reduced.

     

/

返回文章
返回