TinyDet:轻量级通用检测器中的精确小目标检测
计算机视觉与模式识别
2023-04-10 v1
摘要
小目标检测要求检测头在图像特征图上扫描大量位置,这对于计算和能耗高效的轻量级通用检测器而言极其困难。为了在有限计算量下精确检测小目标,我们提出了一种计算复杂度极低的两阶段轻量级检测框架,称为 TinyDet。它支持高分辨率特征图以进行密集锚定,从而更好地覆盖小目标;提出了一种用于减少计算量的稀疏连接卷积;增强了骨干网络中的早期阶段特征;并解决了特征不对齐问题,以实现精确的小目标检测。在 COCO 基准上,我们的 TinyDet-M 仅用 991 MFLOPs 即实现了 30.3 AP 和 13.5 AP^s,这是首个在低于 1 GFLOPs 下 AP 超过 30 的检测器;此外,TinyDet-S 和 TinyDet-L 在不同计算限制下也取得了令人满意的性能。
引用
@article{arxiv.2304.03428,
title = {TinyDet: Accurate Small Object Detection in Lightweight Generic Detectors},
author = {Shaoyu Chen and Tianheng Cheng and Jiemin Fang and Qian Zhang and Yuan Li and Wenyu Liu and Xinggang Wang},
journal= {arXiv preprint arXiv:2304.03428},
year = {2023}
}