AugFCOS: Augmented fully convolutional one-stage object detection network

Xiuwei Zhang, Wei Guo, Yinghui Xing, Wenna Wang, Hanlin Yin, Yanning Zhang

Research output: Contribution to journalArticlepeer-review

25 Scopus citations

Abstract

As a pioneering work of introducing the idea of full convolutional network into the field of object detection, the fully convolutional one-stage object detection network (FCOS) has the advantage of excellent performance with low memory overhead. However, there are certain problems with FCOS that merit more research: the centerness quality assessment loss does not decrease during the late training stage, and its adaptive training sample selection (ATSS) relies heavily on the hyperparameter. To solve the aforementioned problems, we propose a novel object detection network, named augmented fully convolutional one-stage object detection network (AugFCOS). First of all, we propose an improved dynamic optimization loss (DOL) to mitigate the impact of the original centerness loss not decreasing. Then, a Robust Training Sample Selection (RTSS) is proposed to get rid of the dependence of hyper-parameter in ATSS of FCOS. Finally, a novel mixed attention feature pyramid network (MAFPN) is presented to enhance the multi-scale representation ability of feature pyramid network (FPN) and further improve the ability of multi-scale detection. The experimental results on MS COCO demonstrate the effectiveness of our proposed AugFCOS, where AugFCOS achieves approximate 2.0% to 2.9% increase when compared with ATSS and FCOS.

Original languageEnglish
Article number109098
JournalPattern Recognition
Volume134
DOIs
StatePublished - Feb 2023

Keywords

  • Attention module
  • Feature pyramid network
  • Object detection
  • Sample selection

Fingerprint

Dive into the research topics of 'AugFCOS: Augmented fully convolutional one-stage object detection network'. Together they form a unique fingerprint.

Cite this