Journal
SYSTEMS SCIENCE & CONTROL ENGINEERING
Volume 9, Issue 1, Pages 314-321Publisher
TAYLOR & FRANCIS LTD
DOI: 10.1080/21642583.2021.1901156
Keywords
Object detection; Tiny YOLOv3; multi-scale prediction; K-means; real-time
Categories
Funding
- National Natural Science Foundation [61603220, 61733009]
- Research Fund for the Taishan Scholar Project of Shandong Province of China
- SDUST Young Teachers Teaching Talent Training Plan [BJRC20180503, BJRC20190504]
Ask authors/readers for more resources
The improved Tiny YOLOv3 algorithm leverages K-means clustering, pooling, convolution layers, and upsampling/downsampling techniques to enhance feature fusion and multi-scale fusion, ultimately incorporating complete intersection over union in the loss function for improved detection results. It is lightweight and can be trained on a CPU, meeting the requirements of detection speed and accuracy.
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.
Authors
I am an author on this paper
Click your name to claim this paper and add it to your profile.
Reviews
Recommended
No Data Available