一种有效降低anchor数量的anchor-based目标检测方法

AN ANCHOR-BASED OBJECT DETECTION METHOD FOR EFFECTIVELY REDUCING THE NUMBER OF ANCHORS

  • 摘要: 基于anchor的目标检测方法存在计算量庞大的缺点,提出一种有效降低anchor数量的two-stage解决方案位置估计前置网络(Pre-positionEstimationNetwork,PPENet)。在高斯热图上估计目标位置和类别信息,基于目标位置信息仅生成少量的anchor,最后估计目标形状偏差。该方法将目标位置估计前置,利用位置估计信息有效地减少了anchor数量,降低了计算量。该方案是端到端的,在MSCOCO公开数据集上,最高mAP值达到了50.2,最优FPS为33.1,达到了state-of-the-art水平。实验结果证明了该文方法的有效性。

     

    Abstract: The method of object detection based on anchor has the disadvantage of huge computation. In this work, we propose a two-stage solution to effectively reduce the number of anchors, which is pre-position estimation network (PPENet). We estimated the target location and category information on the Gaussian heatmap, generated only a small number of anchors based on the target location information, and estimated the target shape deviation. This method putted the target position estimation in front, and used the position estimation information to effectively reduce the number of anchors and the amount of computation. This solution was end-to-end. On the MSCOCO public dataset, the maximum AP value reached 50.2, and the optimal FPS was 33.1. The scheme reached the state-of-the-art level. The experimental results prove the effectiveness of the proposed method.

     

/

返回文章
返回