《计算机应用与软件杂志》发表论文赏析
作者:李丹阳, 刘卫光, 强赞霞, 肖顺亮
摘要:自动驾驶中要尽早检测到交通标志及时作出行车决策,其中的交通标志属于小目标。针对当前小目标检测精度低的情况,提出基于YOLOv4改进的交通标志检测算法。主要有以下几部分:改进骨干网络,嵌入融合注意力机制模块,加强对通道和空间信息的关注;将交叉熵损失函数改为Focal Loss损失,解决样本分布不平衡的问题;利用图片的多尺度信息进行特征提取,空洞卷积增加感受野。在TT00k数据集上进行实验,结果表明改进之后的网络模型总mAP提升14.16%,其整体性能胜过其他检测方法。
关键词:交通标志检测, 注意力模块, 损失函数, 多尺度特征