An Object Detection Algorithm with Multi-scale Context Information Based on YOLOv4

Abstract

Most current object detection algorithms have the issue of missing objects due to occlusion. As the great difference of scale between occlusion objects and their integrity is affected, how to reduce the missing rate of occlusion objects has become a challenging problem in object detection algorithms. Aiming at the problem, this paper proposes an object detection algorithm based on multi-scale context information on the basis of the YOLOv4 algorithm framework. Firstly, the object features are extracted through the CSPDarkNet53 feature extraction network. Then, the proposed attention-guided multi-scale context information module (AMCM) enhances the ability to perceive object context information and endows features with different influence factors through attention weighting to improve feature discrimination. The experimental results show that the detection accuracy and recall of the proposed algorithm reach 83.2% and 96.4% respectively on PASCAL VOC dataset, and 83.2% and 91% on KITTI dataset, which improves the detection performance of YOLOv4 algorithm.