首页 | 本学科首页   官方微博 | 高级检索  
     检索      

基于YOLOX-S的车窗状态识别算法
引用本文:黄键,徐伟峰,苏攀,王洪涛,李真真.基于YOLOX-S的车窗状态识别算法[J].吉林大学学报(理学版),2023(4):875-882.
作者姓名:黄键  徐伟峰  苏攀  王洪涛  李真真
作者单位:1. 华北电力大学(保定)计算机系
基金项目:国家自然科学基金(批准号:61802124);;中央高校基本科研业务费专项基金(批准号:2021MS089);
摘    要:通过对YOLOX-S模型引入可变形卷积神经网络和焦点损失函数(Focal loss),解决原YOLOX-S模型车窗识别准确率较低的问题.首先,通过在YOLOX-S模型的主干特征提取网络中引入可变形卷积神经网络,对卷积核中的各采样点引入偏移量,以便在原始图像中提取到更具有表征的信息,从而提高车窗识别的精准度;其次,使用Focal loss替代原模型中的二元交叉熵损失函数,Focal loss能缓解正负样本不平衡对训练的影响,其在训练过程中更关注难样本,从而提高了模型对车窗目标的识别性能;最后,为验证改进算法的性能,实验收集并标注15 627张图片进行训练和验证.实验结果表明,改进后的车窗识别算法的平均目标精度提高了3.88%.

关 键 词:车窗识别  YOLOX-S模型  可变形卷积神经网络  焦点损失
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号