Lightweight Semantic Segmentation for RoboCup SPL: A U-Net Model with MobileNetV2 and Attention Mechanisms
Real-time and precise semantic segmentation is essential for robotic vision systems, particularly in dynamic and complex environments such as RoboCup SPL. This paper introduces a lightweight semantic segmentation model that combines the U-Net architecture with a MobileNetV2 subnetwork and the Convolutional Block Attention Module (CBAM). The UNet's encoder-decoder structure facilitates effective multi-scale feature fusion, while MobileNetV2 reduces computational overhead. CBAM further enhances the model's ability to focus on critical features. Evaluations on the RoboCup SPL instance segmentation dataset achieved a mean accuracy of 0.9370 . Through comparative analysis with various backbone networks and ablation studies, we investigated the effects of CBAM, joint loss functions, and data augmentation on model performance. Results indicate that the CBAM mechanism significantly improves segmentation accuracy, and both joint loss functions and data augmentation offer additional benefits. The proposed model achieves a favorable balance between accuracy and computational efficiency, making it highly suitable for real-time applications.
Authors
- Jinwei Xia (ORCID: https://orcid.org/0009-0005-3926-7532)
- Ruiyang Su
Institutions
- Jinling Institute of Technology (CN)
- Shanghai Ocean University (CN)
Publication Details
- Published
- 2024-09-13
- DOI
- https://doi.org/10.1109/aiotc63215.2024.10748302
- Primary Topic
- Robotics and Automated Systems
- Type
- article
- Field-Weighted Citation Impact
- 0.00