FCOS Small Target Detection Algorithm Combined with Multi-Layer Hybrid Attention Mechanism

Ying Liu, Luyao Geng, Hao Yu, Zhijie Xu

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

Abstract

The current target detection algorithm can be competent for most of the detection tasks. However, improving the detection accuracy of small targets is difficult due to the small target occupy less pixels and the feature extraction is hard to achieve. To address this problem, the per-pixel target detection algorithm FCOS is adapted in this research, and the widely used ResNet50 is implemented as the algorithm backbone, by adjusting the size of the input image and the composition of the loss function. The CBAM hybrid attention mechanism is applied into the shallow features and high-level features corresponding to the bottom pyramid, then the feature pyramid is constructed to achieve the purpose of multi-scale target detection. The comparison and ablation experiments show that the original FCOS target detection model can improve the detection accuracy by about 3.7% and the small target detection accuracy by about 2% on the MS-COCO dataset.

Original languageEnglish
Title of host publicationAIPR 2021 - 2021 4th International Conference on Artificial Intelligence and Pattern Recognition
PublisherAssociation for Computing Machinery (ACM)
Pages50-55
Number of pages6
ISBN (Electronic)9781450384087
DOIs
Publication statusPublished - 24 Sep 2021
Event4th International Conference on Artificial Intelligence and Pattern Recognition - Virtual, Online, China
Duration: 24 Sep 202126 Sep 2021
Conference number: 4
https://dl.acm.org/doi/proceedings/10.1145/3488933

Conference

Conference4th International Conference on Artificial Intelligence and Pattern Recognition
Abbreviated titleAIPR 2021
Country/TerritoryChina
CityVirtual, Online
Period24/09/2126/09/21
Internet address

Fingerprint

Dive into the research topics of 'FCOS Small Target Detection Algorithm Combined with Multi-Layer Hybrid Attention Mechanism'. Together they form a unique fingerprint.

Cite this