Multi-Object Detection and Classification in Construction Sites Based on YOLOv5
Xintao Liang, Tianqi Jiang, Qiang Fu, Qingyan Wang · 2023
A thorough analysis of the framework structure of the YOLO algorithm is conducted, and based on the YOLOv5 algorithm, rapid detection and classification of extracted features are implemented. Addressing the issue of multi-object detection and classification in engineering sites, this study utilizes the YOLOv5 algorithm for object detection in engineering scenarios, constructs the ResNet50 network, and achieves training and recognition of categories for engineering vehicles, whether washed or unwashed. Additionally, the YOLOv5 algorithm is employed to detect whether construction personnel are wearing safety helmets and reflective clothing, enabling fast and accurate entity detection and classification at construction sites.