Object detection technology is pivotal for autonomous vehicles to perceive their surroundings, focusing on static road traffic objects like traffic lights, signs, and pavement markings. Training separate networks for each detection task is resource-inefficient under constrained conditions. In order to address this issue, our research introduces the UniDet model, a unified framework for multi-task object detection. It efficiently handles multiple tasks with resource usage and inference speeds close to those of single-task models, outperforming the baseline by 7.1% in Average Precision (AP) on traffic datasets. We observed that multi-task detection often faces data and feature conflicts. To address these challenges, we developed a task decoupling strategy for collaborative cross-task learning. This mitigates learning conflicts due to inadequate data annotations and resolves feature conflicts between classification and localization, leading to a 2.2% performance boost. Additionally, the prevalent issue of detecting small objects was addressed by introducing multi-scale feature fusion and loss functions. A global context-aware aggregation module enhances feature extraction for small objects, while an instance scale-aware weight redistribution strategy tackles the learning imbalance, cumulatively increasing the model’s AP by 1.4% and specifically by 3.1% for small objects. This streamlined approach ensures methodological rigor and emphasizes the model’s innovations and contributions to the field of autonomous vehicle perception.

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

UniDet: A Unified Multi-head Approach for Enhanced Detection of Static Road Traffic Targets

  • Xiubo Liang,
  • Hongzhi Wang,
  • Bo Zhang,
  • Jinxing Han,
  • Tanghu Feng,
  • Qifei Zhang

摘要

Object detection technology is pivotal for autonomous vehicles to perceive their surroundings, focusing on static road traffic objects like traffic lights, signs, and pavement markings. Training separate networks for each detection task is resource-inefficient under constrained conditions. In order to address this issue, our research introduces the UniDet model, a unified framework for multi-task object detection. It efficiently handles multiple tasks with resource usage and inference speeds close to those of single-task models, outperforming the baseline by 7.1% in Average Precision (AP) on traffic datasets. We observed that multi-task detection often faces data and feature conflicts. To address these challenges, we developed a task decoupling strategy for collaborative cross-task learning. This mitigates learning conflicts due to inadequate data annotations and resolves feature conflicts between classification and localization, leading to a 2.2% performance boost. Additionally, the prevalent issue of detecting small objects was addressed by introducing multi-scale feature fusion and loss functions. A global context-aware aggregation module enhances feature extraction for small objects, while an instance scale-aware weight redistribution strategy tackles the learning imbalance, cumulatively increasing the model’s AP by 1.4% and specifically by 3.1% for small objects. This streamlined approach ensures methodological rigor and emphasizes the model’s innovations and contributions to the field of autonomous vehicle perception.