arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2011.04214cs.CV

An improved helmet detection method for YOLOv3 on an unbalanced dataset

Dalian University of Technology(大连理工大学);Beijing International Studies University(北京第二外国语学院)

Rui Geng, Yixuan Ma, Wanhong Huang

英文摘要:

The YOLOv3 target detection algorithm is widely used in industry due to its high speed and high accuracy, but it has some limitations, such as the accuracy degradation of unbalanced datasets. The YOLOv3 target detection algorithm is based on a Gaussian fuzzy data augmentation approach to pre-process the data set and improve the YOLOv3 target detection algorithm. Through the efficient pre-processing, the confidence level of YOLOv3 is generally improved by 0.01-0.02 without changing the recognition speed of YOLOv3, and the processed images also perform better in image localization due to effective feature fusion, which is more in line with the requirement of recognition speed and accuracy in production.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑