YOLC: You Only Look Clusters for Tiny Object Detection in Aerial Images

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Liu, Chenguang, Gao, Guangshuai, Huang, Ziyue, Hu, Zhenghui, Liu, Qingjie, Wang, Yunhong
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866916289537114112
author Liu, Chenguang
Gao, Guangshuai
Huang, Ziyue
Hu, Zhenghui
Liu, Qingjie
Wang, Yunhong
author_facet Liu, Chenguang
Gao, Guangshuai
Huang, Ziyue
Hu, Zhenghui
Liu, Qingjie
Wang, Yunhong
contents Detecting objects from aerial images poses significant challenges due to the following factors: 1) Aerial images typically have very large sizes, generally with millions or even hundreds of millions of pixels, while computational resources are limited. 2) Small object size leads to insufficient information for effective detection. 3) Non-uniform object distribution leads to computational resource wastage. To address these issues, we propose YOLC (You Only Look Clusters), an efficient and effective framework that builds on an anchor-free object detector, CenterNet. To overcome the challenges posed by large-scale images and non-uniform object distribution, we introduce a Local Scale Module (LSM) that adaptively searches cluster regions for zooming in for accurate detection. Additionally, we modify the regression loss using Gaussian Wasserstein distance (GWD) to obtain high-quality bounding boxes. Deformable convolution and refinement methods are employed in the detection head to enhance the detection of small objects. We perform extensive experiments on two aerial image datasets, including Visdrone2019 and UAVDT, to demonstrate the effectiveness and superiority of our proposed approach. Code is available at https://github.com/dawn-ech/YOLC.
format Preprint
id arxiv_https___arxiv_org_abs_2404_06180
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle YOLC: You Only Look Clusters for Tiny Object Detection in Aerial Images
Liu, Chenguang
Gao, Guangshuai
Huang, Ziyue
Hu, Zhenghui
Liu, Qingjie
Wang, Yunhong
Computer Vision and Pattern Recognition
Detecting objects from aerial images poses significant challenges due to the following factors: 1) Aerial images typically have very large sizes, generally with millions or even hundreds of millions of pixels, while computational resources are limited. 2) Small object size leads to insufficient information for effective detection. 3) Non-uniform object distribution leads to computational resource wastage. To address these issues, we propose YOLC (You Only Look Clusters), an efficient and effective framework that builds on an anchor-free object detector, CenterNet. To overcome the challenges posed by large-scale images and non-uniform object distribution, we introduce a Local Scale Module (LSM) that adaptively searches cluster regions for zooming in for accurate detection. Additionally, we modify the regression loss using Gaussian Wasserstein distance (GWD) to obtain high-quality bounding boxes. Deformable convolution and refinement methods are employed in the detection head to enhance the detection of small objects. We perform extensive experiments on two aerial image datasets, including Visdrone2019 and UAVDT, to demonstrate the effectiveness and superiority of our proposed approach. Code is available at https://github.com/dawn-ech/YOLC.
title YOLC: You Only Look Clusters for Tiny Object Detection in Aerial Images
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2404.06180