An Approach on Image Processing of Deep Learning Based on Improved SSD

被引:28
作者
Jin, Liang [1 ]
Liu, Guodong [1 ]
机构
[1] Harbin Inst Technol, Harbin 150001, Peoples R China
来源
SYMMETRY-BASEL | 2021年 / 13卷 / 03期
关键词
deep learning; object detection; remote sensing images; ship detection; SSD; OBJECT DETECTION; SHIP DETECTION; NETWORK;
D O I
10.3390/sym13030495
中图分类号
O [数理科学和化学]; P [天文学、地球科学]; Q [生物科学]; N [自然科学总论];
学科分类号
07 ; 0710 ; 09 ;
摘要
Compared with ordinary images, each of the remote sensing images contains many kinds of objects with large scale changes, providing more details. As a typical object of remote sensing image, ship detection has been playing an essential role in the field of remote sensing. With the rapid development of deep learning, remote sensing image detection method based on convolutional neural network (CNN) has occupied a key position. In remote sensing images, the objects of which small scale objects account for a large proportion are closely arranged. In addition, the convolution layer in CNN lacks ample context information, leading to low detection accuracy for remote sensing image detection. To improve detection accuracy and keep the speed of real-time detection, this paper proposed an efficient object detection algorithm for ship detection of remote sensing image based on improved SSD. Firstly, we add a feature fusion module to shallow feature layers to refine feature extraction ability of small object. Then, we add Squeeze-and-Excitation Network (SE) module to each feature layers, introducing attention mechanism to network. The experimental results based on Synthetic Aperture Radar ship detection dataset (SSDD) show that the mAP reaches 94.41%, and the average detection speed is 31FPS. Compared with SSD and other representative object detection algorithms, this improved algorithm has a better performance in detection accuracy and can realize real-time detection.
引用
收藏
页数:15
相关论文
共 44 条
  • [1] [Anonymous], CoRR
  • [2] Bochkovskiy A., 2020, PREPRINT, DOI DOI 10.48550/ARXIV.2004.10934
  • [3] A Unified Multi-scale Deep Convolutional Neural Network for Fast Object Detection
    Cai, Zhaowei
    Fan, Quanfu
    Feris, Rogerio S.
    Vasconcelos, Nuno
    [J]. COMPUTER VISION - ECCV 2016, PT IV, 2016, 9908 : 354 - 370
  • [4] A Deep Neural Network Based on an Attention Mechanism for SAR Ship Detection in Multiscale and Complex Scenarios
    Chen, Chen
    He, Chuan
    Hu, Changhua
    Pei, Hong
    Jiao, Licheng
    [J]. IEEE ACCESS, 2019, 7 : 104848 - 104863
  • [5] Fall Detection Based on Key Points of Human-Skeleton Using OpenPose
    Chen, Weiming
    Jiang, Zijie
    Guo, Hailin
    Ni, Xiaoyang
    [J]. SYMMETRY-BASEL, 2020, 12 (05):
  • [6] From Feature to Paradigm: Deep Learning in Machine Translation
    Costa-Jussa, Marta R.
    [J]. JOURNAL OF ARTIFICIAL INTELLIGENCE RESEARCH, 2018, 61 : 947 - 974
  • [7] Dai JF, 2016, ADV NEUR IN, V29
  • [8] Histograms of oriented gradients for human detection
    Dalal, N
    Triggs, B
    [J]. 2005 IEEE COMPUTER SOCIETY CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, VOL 1, PROCEEDINGS, 2005, : 886 - 893
  • [9] A light and faster regional convolutional neural network for object detection in optical remote sensing images
    Ding, Peng
    Zhang, Ye
    Deng, Wei-Jian
    Jia, Ping
    Kuijper, Arjan
    [J]. ISPRS JOURNAL OF PHOTOGRAMMETRY AND REMOTE SENSING, 2018, 141 : 208 - 218
  • [10] Fayjie AR, 2018, INT CONF UBIQ ROBOT, P896