Müdigkeitserkennung beim Fahren basierend auf verbesserter YOLOv8n-Pose

CAI Zhongqi ,  

LIN Shanling ,  

LIN Jianpu ,  

LÜ Shanhong ,  

LIN Zhixian ,  

GUO Tailiang ,  

摘要

Angesichts der aktuellen Probleme bei Algorithmen zur Erkennung von Fahrermüdigkeit, wie z. B. komplexe Erkennungsprozesse, viele Parameter, niedrige Genauigkeit und langsame Laufgeschwindigkeit, wurde ein leichtgewichtiges Modell basierend auf einer verbesserten YOLOv8n-Pose vorgeschlagen. Dieses Modell optimiert die Struktur von YOLOv8n-Pose. Erstens wurde im Backbone-Netzwerk des Modells eine Ghost-Convolution eingeführt, um die Anzahl der Modellparameter und unnötige Konvolutionsberechnungen zu reduzieren. Zweitens wurde ein Slim-neck eingeführt, der Merkmale verschiedener Größen aus dem Backbone-Netzwerk fusioniert und die Netzwerkvorhersagebeschleunigung unterstützt. Gleichzeitig wurde im Neck-Netzwerk ein occlusion-aware Aufmerksamkeitmodul (SEAM) hinzugefügt, das den Gesichtsbereich im Bild betont und den Hintergrund abschwächt, um die Genauigkeit der Schlüsselpunktlokalisierung zu verbessern. Schließlich wurde im Detektionskopf eine GNSC-Head-Struktur vorgeschlagen, die Shared Convolution einführt und die traditionelle BN-Schicht in eine stabilere GN-Schicht optimiert, wodurch der Parameterraum und die Rechenressourcen des Modells effektiv eingespart werden. Experimentelle Ergebnisse zeigen, dass die verbesserte YOLOv8n-Pose im Vergleich zum Originalalgorithmus den mAP@0.5 um 0,9 % erhöht, die Anzahl der Parameter und den Rechenaufwand jeweils um 50 % reduziert, die FPS um 8 % erhöht und die finale Müdigkeitserkennung des Fahrers 93,5 % erreicht. Es wurde bestätigt, dass der vorgestellte Algorithmus trotz Leichtgewichtigkeit eine hohe Erkennungsgenauigkeit beibehält und den Fahrzustand effektiv erkennen kann, was eine starke Unterstützung für den Einsatz auf Edge-Geräten in Fahrzeugen bietet.

关键词

Müdigkeitserkennung beim Fahren;Deep Learning;YOLOv8n-Pose;Leichtgewicht;Aufmerksamkeitsmechanismus

阅读全文

以上内容由讯飞翻译自动生成,翻译内容仅供参考。对于因使用本网站翻译内容产生的相关后果,本网站不承担任何商业和法律责任。

The above content is generated by Large Model Translation. The translated content is for reference only. We do not assume any commercial or legal responsibilty for any consequences arising from the use of our website