Real-Time Vehicle Detection from Short-range Aerial Image with Compressed MobileNet

被引:0
作者
He, Yuhang [1 ]
Pan, Ziyu [1 ]
Li, Lingxi [2 ]
Shan, Yunxiao [1 ]
Cao, Dongpu [3 ]
Chen, Long [1 ]
机构
[1] Sun Yat Sen Univ, Sch Data & Comp Sci, Guangzhou, Guangdong, Peoples R China
[2] Indiana Univ Purdue Univ, Purdue Sch Engn & Technol, Dept Elect & Comp Engn, Indianapolis, IN 46202 USA
[3] Univ Waterloo, Dept Mech & Mechatron Engn, Waterloo, ON, Canada
来源
2019 INTERNATIONAL CONFERENCE ON ROBOTICS AND AUTOMATION (ICRA) | 2019年
基金
国家重点研发计划; 中国国家自然科学基金;
关键词
D O I
10.1109/icra.2019.8793673
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Vehicle detection from short-range aerial image faces challenges including vehicle blocking, irrelevant object interference, motion blurring, color variation etc., leading to the difficulty to achieve high detection accuracy and real-time detection speed. In this paper, benefiting from the recent development in MobileNet family network engineering, we propose a compressed MobileNet which is not only internally resistant to the above listed challenges but also gains the best detection accuracy/speed tradeoff when comparing with the original MobileNet. In a nutshell, we reduce the bottleneck architecture number during the feature map downsampling stage but add more bottlenecks during the feature map plateau stage, neither extra FLOPs nor parameters are thus involved but reduced inference time and better accuracy are expected. We conduct experiment on our collected 5-k short-range aerial images, containing six vehicle categories: truck, car, bus, bicycle, motorcycle, crowded bicycles and crowded motorcycles. Our proposed compressed MobileNet achieves 110 FPS (GPU), 31 FPS (CPU) and 15 FPS (mobile phone), 1.2 times faster and 2% more accurate (mAP) than the original MobileNet.
引用
收藏
页码:8339 / 8345
页数:7
相关论文
共 36 条
[1]  
Abadi M., 2015, TENSORFLOW LARGE SCA
[2]  
[Anonymous], INT J COMPUTER VISIO
[3]  
[Anonymous], 2014, ECCV
[4]   Online Photometric Calibration of Auto Exposure Video for Realtime Visual Odometry and SLAM [J].
Bergmann, Paul ;
Wang, Rui ;
Cremers, Daniel .
IEEE ROBOTICS AND AUTOMATION LETTERS, 2018, 3 (02) :627-634
[5]  
Caraffi C, 2012, IEEE INT C INTELL TR, P975, DOI 10.1109/ITSC.2012.6338748
[6]  
Chen L., 2017, T INTELLIGENT TRANSP
[7]   Xception: Deep Learning with Depthwise Separable Convolutions [J].
Chollet, Francois .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :1800-1807
[8]  
Courbariaux M., 2016, ARXIV160202830
[9]  
Dai J., 2016, ADV NEURAL INFORM PR, P379, DOI DOI 10.1109/CVPR.2017.690
[10]  
Deng J., 2019, CVPR