-
题名基于YOLOX-S的车窗状态识别算法
被引量:2
- 1
-
-
作者
黄键
徐伟峰
苏攀
王洪涛
李真真
-
机构
华北电力大学(保定)计算机系
河北省能源电力知识计算重点实验室
-
出处
《吉林大学学报(理学版)》
CAS
北大核心
2023年第4期875-882,共8页
-
基金
国家自然科学基金(批准号:61802124)
全国高等院校计算机基础教育研究会项目(批准号:2019-AFCEC-125)
中央高校基本科研业务费专项基金(批准号:2021MS089).
-
文摘
通过对YOLOX-S模型引入可变形卷积神经网络和焦点损失函数(Focal loss),解决原YOLOX-S模型车窗识别准确率较低的问题.首先,通过在YO LOX-S模型的主干特征提取网络中引入可变形卷积神经网络,对卷积核中的各采样点引入偏移量,以便在原始图像中提取到更具有表征的信息,从而提高车窗识别的精准度;其次,使用Focal loss替代原模型中的二元交叉熵损失函数,Focal loss能缓解正负样本不平衡对训练的影响,其在训练过程中更关注难样本,从而提高了模型对车窗目标的识别性能;最后,为验证改进算法的性能,实验收集并标注15627张图片进行训练和验证.实验结果表明,改进后的车窗识别算法的平均目标精度提高了3.88%.
-
关键词
车窗识别
yolox-s模型
可变形卷积神经网络
焦点损失
-
Keywords
car window recognition
yolox-s model
deformable convolutional neural network
Focal loss
-
分类号
TP391
[自动化与计算机技术—计算机应用技术]
-