← 返回资源分享
Light-Head R-CNN: In Defense of Two-Stage Object Detector
论文
论文
发布时间2017-11-20
发表arXiv:1711.07264
作者:Gang Yu,Jian Sun,Xiangyu Zhang,Zeming Li,Chao Peng,Yangdong Deng
详细介绍
In this paper, we first investigate why typical two-stage methods are not as
fast as single-stage, fast detectors like YOLO and SSD. We find that Faster
R-CNN and R-FCN perform an intensive computation after or before RoI warping.
Faster R-CNN involves two fully connected layers for RoI recognition, while
R-FCN produces a large score maps. Thus, the speed of these networks is slow
due to the heavy-head design in the architecture. Even if we significantly
reduce the base model, the computation cost cannot be largely decreased
accordingly.
We propose a new two-stage detector, Light-Head R-CNN, to address the
shortcoming in current two-stage approaches. In our design, we make the head of
network as light as possible, by using a thin feature map and a cheap R-CNN
subnet (pooling and single fully-connected layer). Our ResNet-101 based
light-head R-CNN outperforms state-of-art object detectors on COCO while
keeping time efficiency. More importantly, simply replacing the backbone with a
tiny network (e.g, Xception), our Light-Head R-CNN gets 30.7 mmAP at 102 FPS on
COCO, significantly outperforming the single-stage, fast detectors like YOLO
and SSD on both speed and accuracy. Code will be made publicly available.
代码仓库 (6)
rickyHong/pytorch-light-head-rcnn-replPyTorch
Stick-To/LH-RCNN-tensorflowTensorFlow
Stick-To/light-head-rcnn-tensorflowTensorFlow
makefile/frcnn
princefr/Light-Head.pytorchPyTorch
reeered/light-headMindSpore
