FCOSR：一种用于航拍目标检测的简单无锚旋转检测器
计算机视觉与模式识别
2021-12-02 v2
摘要
现有的基于锚点的有向目标检测方法已取得惊人结果，但这些方法需要一些手动预设框，从而引入了额外的超参数与计算量。现有的无锚方法通常架构复杂且不易部署。我们的目标是提出一种简单且易部署的航拍图像检测算法。本文提出一种基于 FCOS 的单阶段无锚旋转目标检测器（FCOSR），可部署于大多数平台。FCOSR 具有仅由卷积层组成的简单架构。我们的工作聚焦于训练阶段的标签分配策略。我们采用椭圆中心采样方法为定向边界框（OBB）定义合适的采样区域。模糊样本分配策略为重叠对象提供了合理的标签。为解决采样不足问题，设计了多级采样模块。这些策略为训练样本分配了更合适的标签。我们的算法在 DOTA1.0、DOTA1.5 和 HRSC2016 数据集上分别取得了 79.25、75.41 和 90.15 的 mAP。FCOSR 在单尺度评估中展现出优于其他方法的性能。我们将轻量版 FCOSR 模型转换为 TensorRT 格式，在 Jetson Xavier NX 上以单尺度达到 10.68 FPS 的速度取得 DOTA1.0 上 73.93 的 mAP。代码见：https://github.com/lzh420202/FCOSR
引用
@article{arxiv.2111.10780,
title = {FCOSR: A Simple Anchor-free Rotated Detector for Aerial Object Detection},
author = {Zhonghua Li and Biao Hou and Zitong Wu and Licheng Jiao and Bo Ren and Chen Yang},
journal= {arXiv preprint arXiv:2111.10780},
year = {2021}
}
备注
10 pages, 6 tables, 7 figures