YOLO is a milestone algorithm of object detection, which is the first One-stage detector in deep learning era. In spite of its great improvement of detection speed, the detection accuracy is somewhat insufficient, especially for small targets. In this paper, U-shaped module based on YOLOv4 (U-YOLO) is proposed. First, multi-level features extracted by CSPDarknet using Feature Pyramid Network (FPN) are fused. Then, the fused features is fed into multiple U-shaped modules. Finally, feature maps consisting of the features from different U-shaped modules are gathered up to construct a feature pyramid for object detection. Experiment shows that the U-shaped module can improve the accuracy of YOLOv4.
|