arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2012.01724cs.CV

用于精确单阶段目标检测的并行残差双向融合特征金字塔网络

Parallel Residual Bi-Fusion Feature Pyramid Network for Accurate Single-Shot Object Detection

National Yang Ming Chiao Tung University(国立阳明交通大学);University at Albany, SUNY(纽约州立大学奥尔巴尼分校)

Ping-Yang Chen, Ming-Ching Chang, Jun-Wei Hsieh, Yong-Sheng Chen

AI总结:

本文提出PRB-FPN网络,通过引入并行双向融合结构、CORE模块及残差设计Re-CORE,解决了特征金字塔定位不准及难同时检测大小目标的问题,在UAVDT17和MS COCO数据集上实现了高精度单阶段目标检测。

摘要:

本文提出了并行残差双向融合特征金字塔网络(PRB-FPN),用于快速且准确的单阶段目标检测。特征金字塔(FP)在近期的视觉检测中被广泛使用,然而由于池化偏移,FP的自顶向下路径无法保持准确的定位。随着使用层数更多的深层骨干网络,FP的优势被削弱。此外,它无法同时保持对小型和大型目标的准确检测。为了解决这些问题,我们提出了一种具有双向(自顶向下和自底向上)融合的新型并行FP结构及相关改进,以保留高质量特征从而实现精确定位。我们提供了以下设计改进:(1)带有自底向上融合模块(BFM)的并行双向融合FP结构,可同时高精度检测小型和大型目标。(2)拼接与重组(CORE)模块为特征融合提供了自底向上路径,从而构成双向融合FP,能够从较低层特征图中恢复丢失的信息。(3)CORE特征被进一步提纯以保留更丰富的上下文信息。这种在自顶向下和自底向上路径中的CORE提纯仅需少数几次迭代即可完成。(4)向CORE添加残差设计形成了新的Re-CORE模块,能够轻松训练并与各种更深层或更轻量的骨干网络集成。所提出的网络在UAVDT17和MS COCO数据集上取得了最先进的性能。代码可在https://github.com/pingyang1117/PRBNet_PyTorch获取。

英文摘要:

This paper proposes the Parallel Residual Bi-Fusion Feature Pyramid Network (PRB-FPN) for fast and accurate single-shot object detection. Feature Pyramid (FP) is widely used in recent visual detection, however the top-down pathway of FP cannot preserve accurate localization due to pooling shifting. The advantage of FP is weakened as deeper backbones with more layers are used. In addition, it cannot keep up accurate detection of both small and large objects at the same time. To address these issues, we propose a new parallel FP structure with bi-directional (top-down and bottom-up) fusion and associated improvements to retain high-quality features for accurate localization. We provide the following design improvements: (1) A parallel bifusion FP structure with a bottom-up fusion module (BFM) to detect both small and large objects at once with high accuracy. (2) A concatenation and re-organization (CORE) module provides a bottom-up pathway for feature fusion, which leads to the bi-directional fusion FP that can recover lost information from lower-layer feature maps. (3) The CORE feature is further purified to retain richer contextual information. Such CORE purification in both top-down and bottom-up pathways can be finished in only a few iterations. (4) The adding of a residual design to CORE leads to a new Re-CORE module that enables easy training and integration with a wide range of deeper or lighter backbones. The proposed network achieves state-of-the-art performance on the UAVDT17 and MS COCO datasets. Code is available at https://github.com/pingyang1117/PRBNet_PyTorch.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑