Real-time Object Detection with Attention Mask

Abstract

With the development of deep convolutional neural network, the performance of object detection is obviously improved. However, there are still some challenges such as small size and occlusion. In this paper, we present a novel detector named Attention Mask Detector (AMDet). Our motivation is using mask to enhance foreground features and suppress background ones. The mask is produced by an attention branch which is supervised by weak segmentation ground-truth. This weak segmentation ground-truth is generated by bounding box without extra annotations. Our method is based on one-stage detector. We do experiments on both PASCAL VOC and MS COCO datasets and have a result comparison with other one-stage detectors.