Learning Single-Shot Detector with Mask Prediction and Gate Mechanism

查看全文 作　　者：Jingyi [1]Chen;Haiwei [1]Pan;Qianna [1]Cui;Yang [1]Dong;Shuning [1]He 机构地区：[1]Harbin Engineering University,Harbin,People’s Republic of China 出　　处：《国际计算机前沿大会会议论文集》索引2020年第1期,共12页 基　　金：the National Natural Science Foundation of China under Grant No.61672181。 摘　　要：Detection efficiency plays an increasingly important role in object detection tasks.One-stage methods are widely adopted in real life because of their high efficiency especially in some real-time detection tasks such as face recognition and self-driving cars.RetinaMask achieves significant progress in the field of one-stage detectors by adding a semantic segmentation branch,but it has limitation in detecting multi-scale objects.To solve this problem,this paper proposes RetinaMask with Gate(RMG)model,consisting of four main modules.It develops RetinaMask with a gate mechanism,which extracts and combines features at different levels more effectively according to the size of objects.It firstly extracted multi-level features from input image by ResNet.Secondly,it constructed a fused feature pyramid through feature pyramid network,then gate mechanism was employed to adaptively enhance and integrate features at various scales with the respect to the size of object.Finally,three prediction heads were added for classification,localization and mask prediction,driving the model to learn with mask prediction.The predictions of all levels were integrated during the post-processing.The augment network shows better performance in object detection without the increase of computation cost and inference time,especially for small objects. 关 键 词：Single-shot detector Feature pyramid networks Gate mechanism Mask prediction