YOLOv3 Precision Improvement by the Weighted Centers of Confidence Selection
Adrian Horzyk, Efe Ergun
Abstract
One of the most popular and widely used object detection algorithm today is the YOLOv3 due to its high performance and speed. However, YOLOv3 is not the best algorithm in terms of precision. This paper introduces a substantial change to the post-processing routine of the YOLOv3 after the prediction to increase its final accuracy. Currently, YOLOv3 uses a Non-Max Suppression algorithm to eliminate multiple detections of the same object. This algorithm is picking the most confident overlaying box on any object to present it as the final prediction. This paper presents a new algorithm called Weighted Centers of Confidence Selection that increases the precision using a confidence-weighted average bounding box as a replacement to the existing bounding boxes without making any changes to the YOLOv3 convolutional neural network. We demonstrate how this algorithm works and compare its results to the results achieved by the YOLO's Non-Max Suppression algorithm, focusing on precision and achieving almost the same frame-speed as the original YOLOv3. This new approach allowed us to improve the average accuracy on the COCO dataset in comparison to the original YOLO's Non-Max Suppression.