Two-Stage Object Detection Based on Deep Pruning for Remote Sensing Image
In this paper, we concentrate on tackling the problems of object detection in very-high-resolution (VHR) remote sensing images. The main challenges of object detection in VHR remote sensing images are: (1) VHR images are usually too large and it will consume too much time when locating objects; (2) high false alarm because background dominate and is complex in VHR images. To address the above challenges, a new method is proposed to build two-stage object detection model. Our proposed method can be divided into two processes: (1) we use twice pruning to get region proposal convolutional neural network which is used to predict region proposals; (2) and we use once pruning to get classification convolutional neural network which is used to analyze the result of the first stage and output the class labels of proposals. The experimental results show that the proposed method has high precision and is significantly faster than the state-of-the-art methods on NWPU VHR-10 remote sensing dataset.
KeywordsVery-high-resolution remote sensing image Computer vision Object detection Convolutional neural network Deep learning
This work is supported by the National Natural Science Foundation of China (61472161), Science & Technology Development Project of Jilin Province (20180101334JC), Natural Science Foundation of Hunan Province (No. 2018JJ3479).
- 3.Luo, Q., Shi, Z.: Airplane detection in remote sensing images based on object proposal. In: 2016 IEEE International Geoscience and Remote Sensing Symposium (IGARSS), pp. 1388–1391. IEEE Press (2016)Google Scholar
- 5.Cao, Y., Niu, X., Dou, Y.: Region-based convolutional neural networks for object detection in very high resolution remote sensing images. In: 12th International Conference on Natural Computation, Fuzzy Systems and Knowledge Discovery (ICNC-FSKD), pp. 548–554. IEEE Press (2016)Google Scholar
- 6.Zhang, R., Yao, J., Zhang, K., Feng, C., Zhang, J.: S-CNN ship detection from high-resolution remote sensing images. ISPRS - International Archives of the Photogrammetry, Remote Sensing and Spatial Information Sciences, XLI-B7, pp. 423–430 (2016)Google Scholar
- 9.Shao, W., Yang, W., Liu, G., Liu, J.: Car detection from high-resolution aerial imagery using multiple features. In: 2012 IEEE International Geoscience and Remote Sensing Symposium (IGARSS), pp. 4379–4382. IEEE Press (2012)Google Scholar
- 11.Jiang, Q., Cao, L., Cheng, M., Wang, C., Li, J.: Deep neural networks-based vehicle detection in satellite images. In: 2015 International Symposium on Bioelectronics and Bioinformatics (ISBB), pp. 184–187. IEEE Press (2015)Google Scholar
- 13.Wu, H., Zhang, H., Zhang, J., Xu, F.: Typical target detection in satellite images based on convolutional neural networks. In: 2015 IEEE International Conference on Systems, Man, and Cybernetics (SMC), pp. 2956–2961. IEEE Press (2015)Google Scholar
- 15.Girshick, R., Donahue, J., Darrell, T., Malik, J.: Rich feature hierarchies for accurate object detection and semantic segmentation. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 580–587 (2014)Google Scholar
- 16.Denil, M., Shakibi, B., Dinh, L., De Freitas, N.: Predicting parameters in deep learning. In: Advances in Neural Information Processing Systems, pp. 2148–2156 (2013)Google Scholar
- 20.Krizhevsky, A., Sutskever, I., Hinton, G. E.: Imagenet classification with deep convolutional neural networks. In: Advances in Neural Information Processing Systems, pp. 1097–1105 (2012)Google Scholar