发明公开
- 专利标题: 基于生成对抗网络与知识图表蒸馏的分心驾驶检测方法
-
申请号: CN202310327493.9申请日: 2023-03-30
-
公开(公告)号: CN116386013A公开(公告)日: 2023-07-04
- 发明人: 高尚兵 , 王腾 , 张莹莹 , 李少凡 , 张浩淼 , 赵可钒 , 蒋东山 , 张海艳 , 于坤 , 王媛媛 , 陈晓兵
- 申请人: 淮阴工学院
- 申请人地址: 江苏省淮安市清江浦区枚乘东路1号
- 专利权人: 淮阴工学院
- 当前专利权人: 淮阴工学院
- 当前专利权人地址: 江苏省淮安市清江浦区枚乘东路1号
- 代理机构: 南京苏高专利商标事务所
- 代理商 柏尚春
- 主分类号: G06V20/59
- IPC分类号: G06V20/59 ; G06V10/764 ; G06V10/82 ; G06N3/0475 ; G06N3/094 ; G06N3/096 ; G06N3/0464 ; G06V10/80
摘要:
本发明公开了一种基于生成对抗网络与知识图表蒸馏的分心驾驶检测方法,本发明首先构建GAN模型:生成器网络为改进的MobileNet和判别器网络为yolov5;然后利用教师模型进行知识图表蒸馏:将yolov5的输出作为判别器网络的输出,并将其作为生成器网络的目标输出;生成器网络将通过知识蒸馏的方式来学习判别器网络的输出,以帮助生成器更好地学习生成逼真的样本,同时,学生网络也做目标检测,并提高检测的准确率;在训练过程中,将生成器网络的输出与目标输出进行比较,并使用反向传播算法对生成器网络和判别器网络进行优化。本发明中在StateFarmDistractedDriverDetection数据集上测试,准确率提升了9.7%,可以有效检测出驾驶员分心驾驶状态,提高了驾驶安全性。