IPG-Net: Image Pyramid Guidance Network for Small Object Detection

Ziming Liu1 · Guangyu Gao1,✉ · Lin Sun2 · Li Fang1

1Beijing Institute of Technology · 2Samsung Strategy and Innovation Center

CVPR Workshops 2020

IPG-Net architecture with image pyramid guidance transformation and fusion modules
An image pyramid is transformed and fused into every stage of the backbone stream, so that even the deepest stage retains enough spatial information for bounding box regression.

Abstract

For Convolutional Neural Network-based object detection, there is a typical dilemma: the spatial information is well kept in the shallow layers which unfortunately do not have enough semantic information, while the deep layers have a high semantic concept but lost a lot of spatial information, resulting in serious information imbalance. To acquire enough semantic information for shallow layers, Feature Pyramid Networks (FPN) is used to build a top-down propagated path.

In this paper, except for top-down combining of information for shallow layers, we propose a novel network called Image Pyramid Guidance Network (IPG-Net) to make sure both the spatial information and semantic information are abundant for each layer. IPG-Net has two main parts: the image pyramid guidance transformation module and the image pyramid guidance fusion module. Our main idea is to introduce the image pyramid guidance into the backbone stream to solve the information imbalance problem, which alleviates the vanishment of small object features. The transformation module promises that even in the deepest stage of the backbone there is enough spatial information for bounding box regression and classification.

We applied this network to both one-stage and two-stage detection models, obtaining state-of-the-art results on the most popular benchmarks, MS COCO and Pascal VOC.

Paper

First page of the IPG-Net paper

Ziming Liu, Guangyu Gao, Lin Sun, Li Fang.
IPG-Net: Image Pyramid Guidance Network for Small Object Detection.
IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR) Workshops, 2020.

BibTeX
@inproceedings{liu2020ipg,
  title        = {IPG-Net: Image pyramid guidance network for small object detection},
  author       = {Liu, Ziming and Gao, Guangyu and Sun, Lin and Fang, Li},
  booktitle    = {IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR) Workshops},
  pages        = {1026--1027},
  year         = {2020},
  organization = {IEEE}
}