An improved Tiny YOLOv3 for real-time object detection

• Main Authors: Wendong Gai, Yakun Liu, Jing Zhang, Gang Jing
• Format: Article
• Language: English
• Published: Taylor & Francis Group 2021-01-01
• Series: Systems Science & Control Engineering
• Subjects: object detection; tiny yolov3; multi-scale prediction; k-means; real-time
• Online Access: http://dx.doi.org/10.1080/21642583.2021.1901156

The existing real-time object detection algorithm often omits the objects in the object detection. So an improved Tiny YOLOv3 (you look only once) algorithm is proposed with both lightweight and high accuracy of object detection. The improved Tiny YOLOv3 uses K-means clustering to estimate the size of the anchor boxes for dataset. The pooling and convolution layers are added in the network to strengthen feature fusion and reduce parameters. The network structure increases upsampling and downsampling to enhance multi-scale fusion. The complete intersection over union is added in the loss function, which effectively improves the detection results. In addition, the proposed method has the lightweight module size and can be trained in the CPU. The experimental results show that the proposed method can meet the requirements of the detection speed and accuracy.