基于輕量級主干的YOLOv5駕駛員疲勞檢測算法

摘要： 針對目前基于深度學習的駕駛員疲勞檢測算法存在著參數(shù)量和計算成本較大，難以在算力較低的設備上得到有效應用這一問題，提出一種基于輕量級主干的YOLOv5駕駛員疲勞檢測算法，通過檢測閉眼、張嘴、低頭這3種標簽的時間占比來進行疲勞判斷。算法使用EfficientViT網絡作為模型的主干網絡，降低了整個模型的參數(shù)量以及計算成本，在模型的頸部網絡部分加入上下文變換器模塊并將歸一化沃瑟斯坦距...