本发明公开了一种基于YOLOv5的疲劳状态检测方法,该方法首先采用了Mosaic数据增强方法对原始图像进行增强,再采用注意力机制为后续的特征提取保留更完整的采样信息,最后通过对损失函数进行改进,能够更好的区分检测框和真实框之间的位置关系。现有的疲劳驾驶检测方法大多只采用了单一信号以判断驾驶员疲劳状态,实际行车过程中的信号采集容易受到各种因素干扰,有时采集到的数据不够准确,由于判别依据单一,训练出的模型鲁棒性较差,容易发生误报等情况,反而影响驾驶员的行车安全。本发明利用YOLOv5m网络模型对驾驶员疲劳状态检测。经过验证,本发明在驾驶员疲劳检测中良好的兼顾了精度和速度,证明此方法具有实用性。
第一,本发明通过对损失函数进行改进,能够更好地区分检测框和真实框之间的位置关系。
第二,本发明无论从检测精度、检测速度还是网络参数、权重文件等与YOLO v3,YOLOv4疲劳状态检测方法相比都有一个质的提高。
黑龙江大学(Heilongjiang University),位于黑龙江省哈尔滨市,是黑龙江省人民政府和中华人民共和国教育部、国家国防科技工业局共建的省属综合性大学,黑龙江省“双一流”建设国内一流大学A类高校,入选国家卓越法律人才教育培养计划、中西部高校基础能力建设工程、特色重点学科项目、国家建设高水平大学公派研究生项目、中国政府奖学金来华留学生接收院校、全国深化创新创业教育改革示范高校、教育部来华留学示范基地,是世界翻译教育联盟、中俄新闻教育高校联盟、中俄综合性大学联盟、上海合作组织大学、“一带一路”智库合作联盟成员单位。
评价单位:“科创中国”黑龙江科技服务团 (黑龙江省科学技术协会)
评价时间:2023-11-10
综合评价
技术前景广阔,具备技术成果转移转化要求。
查看更多>