To address the high rates of missed detections and false alarms, as well as the poor robustness in fatigue driving detection for open-pit mine truck drivers, a fatigue driving detection model for mine truck drivers (EBS-YOLO) based on the improved YOLOv8 is constructed to enhance the overall performance of fatigue detection. Firstly, YOLOv8 was used as the basic model for fatigue detection, and a small target detection layer was added to enhance the model's attention to small targets. Secondly, the bottleneck attention module (BAM) was used to improve the model performance to extract small target features, especially eye features. Finally, all cross-stage aggregation modules (C2f) in the backbone network were replaced with efficient multi-scale attention (EMA) modules, thereby effectively reducing model parameters and computational overhead to meet the requirements of a lightweight model. The results showed that the improved YOLOv8 model had a great detection effect with the accuracy, recall rate, and average detection accuracy reaching 93.6%, 93.9%, and 96.5%, respectively, and the memory size of the model was only 4.9 MB. Compared with the YOLOv8 model, the improved model can quickly and accurately identify the fatigue state of mining truck drivers, meet real-time requirements, and effectively prevent fatigue-driving accidents.
| 科 Family | 属数 Number of genus | 种数 Number of species | 占总种数比例 Percentage of total species (%) | 属 Genus | 种数 Number of species | 占总种数比例 Percentage of total species (%) |
|---|---|---|---|---|---|---|
| 鹅膏菌科Amanitaceae | 2 | 11 | 5.26 | 鹅膏菌属 Amanita | 10 | 4.78 |
| 小菇科 Mycenaceae | 2 | 12 | 5.74 | 丝盖伞属 Inocybe | 5 | 2.39 |
| 多孔菌科 Polyporaceae | 8 | 14 | 6.70 | 蜡蘑属 Laccaria | 5 | 2.39 |
| 红菇科 Russulaceae | 3 | 23 | 11.00 | 小皮伞属 Marasmius | 6 | 2.87 |
| 小菇属 Mycena | 11 | 5.26 | ||||
| 光柄菇属 Pluteus | 5 | 2.39 | ||||
| 红菇属 Russula | 17 | 8.13 | ||||
| 栓菌属 Trametes | 5 | 2.39 |