This paper addresses the challenges of vehicle occlusion and insuffi-cient real-time identification in the context of target recognition for autonomous driving scenarios. It proposes a lightweight improved target recognition algo-rithm based on YOLOv8n, named MMW-YOLO. The algorithm employs a more streamlined MobileViTv3 structure, significantly reducing the number of model parameters while maintaining accuracy. Furthermore, it embeds the Multi-scale convolutional attention (MSCA) module to enhance recognition accuracy. Lastly, the Wise-IoU loss function is utilized to boost the model’s generalization. Experiments conducted using the KITTI dataset demonstrate that MMW-YOLO achieves a 5.2% improvement in average precision over the original model and a 53% reduction in parameter count. The experiments verify the superior performance of the MMW-YOLO algorithm for target recognition tasks in autonomous driving scenarios.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

MMW-YOLO: Research on an Improved Vehicle Recognition Algorithm Based on MobileViT

  • Tianyu Liang,
  • Linan Zu

摘要

This paper addresses the challenges of vehicle occlusion and insuffi-cient real-time identification in the context of target recognition for autonomous driving scenarios. It proposes a lightweight improved target recognition algo-rithm based on YOLOv8n, named MMW-YOLO. The algorithm employs a more streamlined MobileViTv3 structure, significantly reducing the number of model parameters while maintaining accuracy. Furthermore, it embeds the Multi-scale convolutional attention (MSCA) module to enhance recognition accuracy. Lastly, the Wise-IoU loss function is utilized to boost the model’s generalization. Experiments conducted using the KITTI dataset demonstrate that MMW-YOLO achieves a 5.2% improvement in average precision over the original model and a 53% reduction in parameter count. The experiments verify the superior performance of the MMW-YOLO algorithm for target recognition tasks in autonomous driving scenarios.