The widespread development of autonomous driving has posed a significant challenge in traffic light detection. Existing detection technologies often fail to effectively balance the real-time performance and accuracy of object detection. This paper proposes a new object detection model, MRT-DETR, based on the DETR series, which effectively achieves a balance between real-time performance and accuracy in traffic light detection. MRT-DETR employs depthwise separable convolutions and optimize the receptive field in certain layers of the backbone network. Also, MRT-DETR replaces the traditional ReLU activation function with the HardSwish activation function. To demonstrate the effectiveness of the MRT-DETR model, this paper conducts experiments on a traffic light detection dataset, comparing it with common object detection models, including CNN-based models and the DETR series, and analyzes the results. The experimental results show that the MRT-DETR model achieves an 80.61% improvement in the \(\text {AP}^{val}\) metric and a 68.42% improvement in the \(\text {AP}^{val}_{50}\) metric compared to the YOLOv10l model. Compared to the RT-DETR model, MRT-DETR improves the \(\text {AP}^{val}\) metric by 3.22% and the \(\text {AP}^{val}_{50}\) metric by 16.93%. In terms of inference time, the MRT-DETR model reduces inference time by 7.8% compared to the RT-DETR model. The experiments show that the MRT-DETR model achieves favorable results in balancing the accuracy and real-time performance of object detection.

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

MRT-DETR: A New Real-Time Object Detection Method

  • Qiming Zhao,
  • Wei Hu,
  • Yueyang Liu,
  • Rujun Yang,
  • Zonghui Wang

摘要

The widespread development of autonomous driving has posed a significant challenge in traffic light detection. Existing detection technologies often fail to effectively balance the real-time performance and accuracy of object detection. This paper proposes a new object detection model, MRT-DETR, based on the DETR series, which effectively achieves a balance between real-time performance and accuracy in traffic light detection. MRT-DETR employs depthwise separable convolutions and optimize the receptive field in certain layers of the backbone network. Also, MRT-DETR replaces the traditional ReLU activation function with the HardSwish activation function. To demonstrate the effectiveness of the MRT-DETR model, this paper conducts experiments on a traffic light detection dataset, comparing it with common object detection models, including CNN-based models and the DETR series, and analyzes the results. The experimental results show that the MRT-DETR model achieves an 80.61% improvement in the \(\text {AP}^{val}\) metric and a 68.42% improvement in the \(\text {AP}^{val}_{50}\) metric compared to the YOLOv10l model. Compared to the RT-DETR model, MRT-DETR improves the \(\text {AP}^{val}\) metric by 3.22% and the \(\text {AP}^{val}_{50}\) metric by 16.93%. In terms of inference time, the MRT-DETR model reduces inference time by 7.8% compared to the RT-DETR model. The experiments show that the MRT-DETR model achieves favorable results in balancing the accuracy and real-time performance of object detection.