用于单阶段目标检测的正负平衡锚框与对齐特征
计算机视觉与模式识别
2019-08-12 v1
摘要
我们提出了一种新颖的单阶段目标检测器，通过抑制易分负样本同时增加正样本来缓解前景-背景类别的不平衡。为此，我们提出了锚框提升模块（APM），该模块预测每个锚框为正样本的概率，并调整其初始位置和形状，以提升正锚框的质量和数量。此外，我们设计了一个高效的特征对齐模块（FAM），借助来自 APM 的位置和形状变换信息，为拟合提升后的锚框提取对齐特征。我们将这两个提出的模块组装到具有编码器-解码器架构的 VGG-16 和 ResNet-101 网络主干上。在 MS COCO 上的大量实验充分表明，我们的模型与替代方法相比具有竞争力（在 test-dev 集上 mAP 为 40.0%），且运行更快（28.6 fps）。
引用
@article{arxiv.1908.03295,
title = {PosNeg-Balanced Anchors with Aligned Features for Single-Shot Object Detection},
author = {Qiankun Tang and Shice Liu and Jie Li and Yu Hu},
journal= {arXiv preprint arXiv:1908.03295},
year = {2019}
}
备注
Submitted to a conference, under review