关键词: 目标检测, YOLO v3算法, GIoU算法, 安全帽佩戴检测

Abstract:

YOLO v3 object detection algorithm has been widely used in industry due to its high speed and high precision. However, there is a problem that the object function is not consistent with the evaluation index. To solve this problem, improved YOLO v3 object detection algorithm is proposed. This algorithm improves the GIoU calculation method and combines it with the YOLO v3 algorithm’s objective function to design a new objective function to achieve IoU local optimization as the local optimization of the objective function. The test results of public dataset VOC2007 and helmet wearing dataset show that the mAP-50 of the improved YOLO v3 increased by 2.07% and 2.05%, respectively, compared with the YOLO v3 algorithm.

Key words: object detection, YOLO v3 algorithm, GIoU algorithm, helmet wearing test