Next Article in Journal
A Modular IoT Hardware Platform for Distributed and Secured Extreme Edge Computing
Previous Article in Journal
Heat Transfer Study in Breast Tumor Phantom during Microwave Ablation: Modeling and Experimental Results for Three Different Antennas
Open AccessArticle

Object Detection Algorithm Based on Improved YOLOv3

Key Laboratory of Modern Power System Simulation and Control & Renewable Energy Technology, Ministry of Education (Northeast Electric Power University), Jilin 132012, China
*
Author to whom correspondence should be addressed.
Electronics 2020, 9(3), 537; https://doi.org/10.3390/electronics9030537
Received: 10 February 2020 / Revised: 22 March 2020 / Accepted: 23 March 2020 / Published: 24 March 2020
(This article belongs to the Special Issue Deep Learning Based Object Detection)
The ‘You Only Look Once’ v3 (YOLOv3) method is among the most widely used deep learning-based object detection methods. It uses the k-means cluster method to estimate the initial width and height of the predicted bounding boxes. With this method, the estimated width and height are sensitive to the initial cluster centers, and the processing of large-scale datasets is time-consuming. In order to address these problems, a new cluster method for estimating the initial width and height of the predicted bounding boxes has been developed. Firstly, it randomly selects a couple of width and height values as one initial cluster center separate from the width and height of the ground truth boxes. Secondly, it constructs Markov chains based on the selected initial cluster and uses the final points of every Markov chain as the other initial centers. In the construction of Markov chains, the intersection-over-union method is used to compute the distance between the selected initial clusters and each candidate point, instead of the square root method. Finally, this method can be used to continually update the cluster center with each new set of width and height values, which are only a part of the data selected from the datasets. Our simulation results show that the new method has faster convergence speed for initializing the width and height of the predicted bounding boxes and that it can select more representative initial widths and heights of the predicted bounding boxes. Our proposed method achieves better performance than the YOLOv3 method in terms of recall, mean average precision, and F1-score. View Full-Text
Keywords: deep learning; object detection; YOLOv3 method; k-means cluster deep learning; object detection; YOLOv3 method; k-means cluster
Show Figures

Figure 1

MDPI and ACS Style

Zhao, L.; Li, S. Object Detection Algorithm Based on Improved YOLOv3. Electronics 2020, 9, 537.

Show more citation formats Show less citations formats
Note that from the first issue of 2016, MDPI journals use article numbers instead of page numbers. See further details here.

Article Access Map by Country/Region

1
Back to TopTop