3 个月前

M3E-Yolo:一种用于交通标志识别的新轻量级网络

M3E-Yolo:一种用于交通标志识别的新轻量级网络

摘要

交通标志识别致力于保障自动驾驶的安全性。受YOLOv5的启发,本文提出一种新模型,旨在解决现有算法在交通标志识别中准确率与效率之间平衡性较差的问题。首先,引入轻量级网络MobileNetV3进行特征提取,有效减少了模型参数量;其次,引入注意力机制模块以增强通道特征,弥补因模型简化导致的准确率下降问题。实验结果表明,本模型在中文交通标志数据集上的mAP值达到93.6%,接近YOLOv5的性能水平,而参数量不足YOLOv5的四分之一,显著提升了模型的效率与实用性。

基准测试

基准方法指标
traffic-sign-detection-on-cctsdb2021M3E-Yolo
mAP@0.5: 93.4

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
M3E-Yolo:一种用于交通标志识别的新轻量级网络 | 论文 | HyperAI超神经