|
作者 | Zhu Yousong1,2 ; Zhao Chaoyang1,2 ; Han Chenxia3; Wang Jinqiao1,2 ; Lu Hanqing1,2
|
出版日期 | 2019
|
会议日期 | 2019-7-8
|
会议地点 | Shanghai, China
|
关键词 | Object Detection
Knowledge Distillation
|
英文摘要 | In this paper, we explore the idea of distilling small networks
for object detection task. More specifically, we propose a twostage approach to learn more compact and efficient detectors
under the single-shot object detection framework by leveraging knowledge distillation. During the 1st stage, we learn the
feature maps of the student model for each of the prediction
head from the teacher model. Instead of fitting the whole feature map directly, here we propose the mask guided structure
including not only the entire feature map (i.e. global features)
but also region features covered by the object (i.e. local features), which can significantly improve the performance of
the student network. For the 2nd stage, the ground-truth is
used to further refine the performance. Experimental results
on PASCAL VOC and KITTI dataset demonstrate the effectiveness of our proposed approach. We achieve 56.88% mAP
on VOC2007 at 143 FPS with the backbone of 1/8 VGG16. |
会议录出版者 | IEEE
|
语种 | 英语
|
源URL | [http://ir.ia.ac.cn/handle/173211/23586]  |
专题 | 自动化研究所_模式识别国家重点实验室_图像与视频分析团队
|
通讯作者 | Zhu Yousong |
作者单位 | 1.Institute of Automation, Chinese Academy of Sciences 2.University of Chinese Academy of Sciences 3.Wuhan University
|
推荐引用方式 GB/T 7714 |
Zhu Yousong,Zhao Chaoyang,Han Chenxia,et al. Mask Guided Knowledge Distillation for Single Shot Detector[C]. 见:. Shanghai, China. 2019-7-8.
|