Helmet Detection Algorithm Based on Single Pixel Zoom

Abstract

The automatic detection technology of helmet wearing is of great significance to the safety monitoring of construction site. It is difficult to detect a helmet when the object is small, overlapped, or unclear. Therefore, we propose SPZ-Det model to improve the performance of detect small targets in helmet detection task. We redesign the backbone network layers and proposes a Single Pixel Zoom (SPZ) module. SPZ-Det introduces the lower-level feature map, while for the same scale feature map, we select the higher-level feature map. Then the important elements of each feature map layer are amplified by SPZ module to solve the problem of small object feature fading. In the attention module of SPZ, the residual mechanism is introduced to weaken the influence of the attention. The proposed model improves the detection accuracy and keeps the detection speed. On the Safety-Helmet-Wearing dataset, the mAP of SPZ-Det is 27.9% higher than Efficientdet-d0 and gets an 8.8% increase compares with YOLOv3.