Night-Time Aerial Image Vehicle Recognition Technology Based on Transfer Learning and Image Enhancement

被引:0
作者
Yuan G. [1 ]
Hou J. [2 ]
Yin K. [1 ]
机构
[1] Signal Processing Department, Nanjing Research Institute of Electronics Technology, Nanjing
[2] College of Electronic and Information, Northwestern Polytechnical University, Xi'an
来源
Jisuanji Fuzhu Sheji Yu Tuxingxue Xuebao/Journal of Computer-Aided Design and Computer Graphics | 2019年 / 31卷 / 03期
关键词
Deep learning; Faster R-CNN algorithm; Night-time aerial image; Transfer learning; Vehicle detection;
D O I
10.3724/SP.J.1089.2019.17320
中图分类号
学科分类号
摘要
In order to identify vehicles in night-time aerial images effectively, this paper proposed an image processing technique based on two-time transfer learning and the Retinex algorithm. It only used a small-scale data set to train the network and then employed a deep learning algorithm based on Faster R-CNN to achieve quick detection of vehicles. Firstly, a transfer learning process was applied between the large-scale ImageNet data set and the mid-scale Chinese Academy of Sciences daytime aerial data set, and then a second transfer learning algorithm was utilized from the day-time mid-scale data set to the night-time small-scale data set. At the same time, the Retinex iterative algorithm was used to process the night-time pictures to enhance their similarity with the day-time pictures, so that the second transfer learning can be effectively performed. The experimental results show that this method can train an effective recognition network on deep learning platforms with small-scale data sets, and its detection performance is superior to the traditional machine learning methods. It also has certain application values in military reconnaissance and traffic control field. © 2019, Beijing China Science Journal Publishing Co. Ltd. All right reserved.
引用
收藏
页码:467 / 473
页数:6
相关论文
共 15 条
[1]  
Jian M., Drone's unique advantages in traffic management, China Public Security, 12, pp. 40-44, (2016)
[2]  
Cheng H.Y., Weng C.C., Chen Y.Y., Vehicle detection in aerial surveillance using dynamic Bayesian networks, IEEE Transactions on Image Processing, 21, 4, pp. 2152-2159, (2012)
[3]  
Freund Y., Schapire R.E., Experiments with a new boosting algorithm, Proceedings of the 13th International Conference on International Conference on Machine Learning, pp. 148-156, (1996)
[4]  
Kim D., Lee D., Myung H., Et al., Object detection and tracking for autonomous underwater robots using weighted template matching, Proceedings of Oceans, pp. 1-5, (2012)
[5]  
Hinz S., Baumgartner A., Vehicle detection in aerial images using generic features, grouping, and context, Proceedings of the 23rd Symposium of the German Association for Pattern Recognition, 2191, pp. 45-52, (2001)
[6]  
Dalal N., Triggs B., Histograms of oriented gradients for human detection, Proceedings of the IEEE Computer Society Conference on Computer Vision and Pattern Recognition, pp. 886-893, (2005)
[7]  
Ergun H., Sert M., Fusing deep convolutional networks for large scale visual concept classification, Proceedings of the 2nd IEEE International Conference on Multimedia Big Data, pp. 210-213, (2016)
[8]  
Zhuang F., Luo P., He Q., Et al., Survey on transfer learning research, Journal of Software, 26, 1, pp. 26-39, (2015)
[9]  
Ren S.Q., He K.M., Girshick R., Et al., Faster R-CNN: towards real-time object detection with region proposal networks, IEEE Transactions on Pattern Analysis and Machine Intelligence, 39, 6, pp. 1137-1149, (2017)
[10]  
Russakovsky O., Deng J., Su H., Et al., ImageNet large scale visual recognition challenge, International Journal of Computer Vision, 115, 3, pp. 211-252, (2015)