基于轻量级主干的YOLOv5驾驶员疲劳检测算法

摘要： 针对目前基于深度学习的驾驶员疲劳检测算法存在着参数量和计算成本较大，难以在算力较低的设备上得到有效应用这一问题，提出一种基于轻量级主干的YOLOv5驾驶员疲劳检测算法，通过检测闭眼、张嘴、低头这3种标签的时间占比来进行疲劳判断。算法使用EfficientViT网络作为模型的主干网络，降低了整个模型的参数量以及计算成本，在模型的颈部网络部分加入上下文变换器模块并将归一化沃瑟斯坦距... （共9页）