| SFE-SSD: Shallow Feature Enhancement SSD for Small Object Detection |
| Received:September 05, 2019 Revised:October 20, 2019 |
| Key Words:
shallow feature enhancement object detection SSD (Single Shot Multibox Detetor) feature fusion strategy
|
| Fund Project:Supported by the National Natural Science Foundation of China (Grant Nos.07002157; U1811463) |
| Author Name | Affiliation | | Hongchen TAN | School of Mathematical Sciences, Dalian University of Technology, Liaoning 116024, P. R. China | | Jun ZHOU | School of Mathematical Sciences, Dalian University of Technology, Liaoning 116024, P. R. China | | Shengjing TIAN | School of Mathematical Sciences, Dalian University of Technology, Liaoning 116024, P. R. China | | Xiuping LIU | School of Mathematical Sciences, Dalian University of Technology, Liaoning 116024, P. R. China |
|
| Hits: 1007 |
| Download times: 628 |
| Abstract: |
| SSD (Single Shot Multibox Detector) is one of the best object detection algorithms with both high accuracy and fast speed, but fails to detect very small size object which lacks enough resolution and enough feature information. In order to solve this problem, the majority of existing methods improve accuracy at the cost of a heavy loss of speed. In this paper, we propose SFE-SSD (Shallow Feature Enhancement SSD) to improve performance of SSD model on small object detection based on a novel and lightweight way of feature enhancement module. Firstly, we apply deconvolution on the shallowest feature map in SSD's feature pyramid to enlarge the feature map size and recover more feature details. Then, we introduce semantic information to the enlarged feature map by multi-scale feature fusion. In addition, SFE-SSD is designed to a parallel network structure, which could reduce loss of speed in some degree. Experimental results show that our approach achieved 78.4$\%$mAP and is higher than baseline SSD by 1.2$\%$ on PASCAL VOC2007, especially with significant improvement on small object detection. The testing speed of SFE-SSD is 81 FPS at the cost of a little loss of speed. |
| Citation: |
| DOI:10.3770/j.issn:2095-2651.2019.06.016 |
| View Full Text View/Add Comment |
|
|
|
