BPH-YOLOv5: improved YOLOv5 based on biformer prediction head for small target cigatette detection
Xiao Zheng, X. Lu · IET conference proceedings. · 2024
Small object detection in the cigarette detection field is a recent popular task. As cigarette targets in surveillance are tiny, the object scale leads to a great challenge of object detection. To solve the issue mentioned above, we propose BPH-YOLOv5 based on YOLOv5. We add one more prediction head to detect different-scale objects, which is more sensitive to tiny objects using multi-scale features. Then we replace the original prediction heads with BiFormer Prediction Heads (BPH) to explore the prediction potential with an attention mechanism. Experiments showed that the mean average precision (mAP) of BPHYOLOv5 in the cigarette dataset reached 90.34%, which was 6.94% higher than that of YOLOv5-s prototype, 7.20% higher than that of YOLOv7, which are encouraging and competitive. Tests on the actual scenarios verified that the accuracy of smallscale target detection was significantly improved.