The design of high-quality feature representation and efficient loss functions can boost the performance of object detection and other computer vision tasks. Currently, object detectors typically utilize FPN to enhance representation capabilities. However, FPN-based methods fail to fully exploit multi-scale features. In this paper, we propose the Adaptive Feature Pyramid Enhancement Network(AFPEN) to address this problem. Specifically, AFPEN consists of the Dilated Context Aggregation(DCA) module and the Weight-Adaptive Channel Integration (WACI) module, which are designed to capture richer contextual information. Besides, existing loss functions struggle to address object scale variations. To mitigate this, this paper introduces the Boundary Alignment IoU(BA-IoU) loss. BA-IoU improves the speed and accuracy of fitting predicted boxes to ground-truth boxes, particularly for objects of varying scales. The final evaluation of our method on COCO 2017 and Pascal VOC datasets shows competitive performance against state-of-the-art FPN-based detectors.

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

ABDet: Adaptive Feature Enhancement and Boundary Alignment Loss for Object Detection

  • Jia Wen,
  • Ting Zhang,
  • Jialin Li,
  • Xinyi Wang

摘要

The design of high-quality feature representation and efficient loss functions can boost the performance of object detection and other computer vision tasks. Currently, object detectors typically utilize FPN to enhance representation capabilities. However, FPN-based methods fail to fully exploit multi-scale features. In this paper, we propose the Adaptive Feature Pyramid Enhancement Network(AFPEN) to address this problem. Specifically, AFPEN consists of the Dilated Context Aggregation(DCA) module and the Weight-Adaptive Channel Integration (WACI) module, which are designed to capture richer contextual information. Besides, existing loss functions struggle to address object scale variations. To mitigate this, this paper introduces the Boundary Alignment IoU(BA-IoU) loss. BA-IoU improves the speed and accuracy of fitting predicted boxes to ground-truth boxes, particularly for objects of varying scales. The final evaluation of our method on COCO 2017 and Pascal VOC datasets shows competitive performance against state-of-the-art FPN-based detectors.