A Ship Rotation Detection Model in Remote Sensing Images Based on Feature Fusion Pyramid Network and Deep Reinforcement Learning

被引:42
作者
Fu, Kun [1 ,2 ]
Li, Yang [1 ,2 ]
Sun, Hao [1 ]
Yang, Xue [1 ,2 ]
Xu, Guangluan [1 ,2 ]
Li, Yuting [3 ]
Sun, Xian [1 ,2 ]
机构
[1] Chinese Acad Sci, Inst Elect, Key Lab Technol Geospatial Informat Proc & Applic, Beijing 100190, Peoples R China
[2] Univ Chinese Acad Sci, Sch Elect Elect & Commun Engn, Beijing 100049, Peoples R China
[3] Beijing Univ Technol, Coll Life Sci & Bioengn, Beijing 100124, Peoples R China
基金
中国国家自然科学基金;
关键词
ship detection; deep reinforcement learning; convolution neural network; feature map fusion;
D O I
10.3390/rs10121922
中图分类号
X [环境科学、安全科学];
学科分类号
08 ; 0830 ;
摘要
Ship detection plays an important role in automatic remote sensing image interpretation. The scale difference, large aspect ratio of ship, complex remote sensing image background and ship dense parking scene make the detection task difficult. To handle the challenging problems above, we propose a ship rotation detection model based on a Feature Fusion Pyramid Network and deep reinforcement learning (FFPN-RL) in this paper. The detection network can efficiently generate the inclined rectangular box for ship. First, we propose the Feature Fusion Pyramid Network (FFPN) that strengthens the reuse of different scales features, and FFPN can extract the low level location and high level semantic information that has an important impact on multi-scale ship detection and precise location of dense parking ships. Second, in order to get accurate ship angle information, we apply deep reinforcement learning to the inclined ship detection task for the first time. In addition, we put forward prior policy guidance and a long-term training method to train an angle prediction agent constructed through a dueling structure Q network, which is able to iteratively and accurately obtain the ship angle. In addition, we design soft rotation non-maximum suppression to reduce the missed ship detection while suppressing the redundant detection boxes. We carry out detailed experiments on the remote sensing ship image dataset, and the experiments validate that our FFPN-RL ship detection model has efficient detection performance.
引用
收藏
页数:26
相关论文
共 39 条
  • [1] [Anonymous], 2018, SENSORS
  • [2] [Anonymous], P 2018 9 IEEE ANN UB
  • [3] [Anonymous], ARXIV160806993
  • [4] [Anonymous], ARXIV160305027
  • [5] [Anonymous], COMPUT SIMUL
  • [6] [Anonymous], P 2018 COMP VIS PATT
  • [7] [Anonymous], 2013, Playing atari with deep reinforcement learning
  • [8] [Anonymous], P INT C MACH LEARN
  • [9] [Anonymous], 2016, INT C LEARNING REPRE
  • [10] [Anonymous], IEEE T PATTERN ANAL