基于角度敏感的空间注意力机制的轻量型旋转目标检测器

尹开石; 杨萌; 顾曦; 王志成

doi:10.11959/j.issn.2096-6652.202133

您当前的位置：

首页 >

文章列表页 >

基于角度敏感的空间注意力机制的轻量型旋转目标检测器

专刊：目标智能检测与识别 | 更新时间：2024-06-05

- 基于角度敏感的空间注意力机制的轻量型旋转目标检测器
- Light weight rotating object detector based on angle sensitive spatial attention mechanism
- 智能科学与技术学报 2021年3卷第3期页码：322-333
- 作者机构：
  
  1. 东华大学计算机科学与技术学院，上海 201620
  2. 中国舰船研究设计中心，湖北武汉 430064
  3. 同济大学电子与信息工程学院，上海 201804
- 作者简介：
  
  [ "尹开石（2001− ），男，东华大学计算机科学与技术学院在读，主要研究方向为图像分析、机器学习" ]
  [ "杨萌（1986− ），男，博士，中国舰船研究设计中心高级工程师，主要研究方向为舰船总体技术、环境智能感知" ]
  [ "顾曦（1996− ），女，同济大学电子与信息工程学院硕士生，主要研究方向为机器学习、深度学习" ]
  [ "王志成（1975− ），男，博士，同济大学电子与信息工程学院副研究员，主要研究方向为机器学习、深度学习" ]
- 基金信息：
  
  国防基础科研计划资助项目;The National Defense Basic Research Program of China(JCKY2020206B037)
- DOI：10.11959/j.issn.2096-6652.202133
  中图分类号： TP181
- 网络出版日期：2021-09，
  
  纸质出版日期：2021-09-15
- 稿件说明：
移动端阅览
尹开石, 杨萌, 顾曦, 等. 基于角度敏感的空间注意力机制的轻量型旋转目标检测器[J]. 智能科学与技术学报, 2021,3(3):322-333.

Kaishi YIN, Meng YANG, Xi GU, et al. Light weight rotating object detector based on angle sensitive spatial attention mechanism[J]. Chinese journal of intelligent science and technology, 2021, 3(3): 322-333.
尹开石, 杨萌, 顾曦, 等. 基于角度敏感的空间注意力机制的轻量型旋转目标检测器[J]. 智能科学与技术学报, 2021,3(3):322-333. DOI： 10.11959/j.issn.2096-6652.202133.

Kaishi YIN, Meng YANG, Xi GU, et al. Light weight rotating object detector based on angle sensitive spatial attention mechanism[J]. Chinese journal of intelligent science and technology, 2021, 3(3): 322-333. DOI： 10.11959/j.issn.2096-6652.202133.

摘要

随着深度学习的快速发展，近年来越来越多的基于锚框的目标检测算法被应用于遥感图像上，然而提高算法精度的代价是牺牲了检测速度。因此，选择无锚框的目标检测网络框架，针对遥感场景的特点，提出了一种旋转框的遥感检测算法。根据旋转框与其外接矩形框的空间位置关系，提出了一种简单有效的旋转框表示方式。此外，设计了一种用于辅助检测旋转目标的角度敏感的空间注意力机制，通过引入角度信息提升模型对旋转目标的检测能力。提出的算法在公开遥感数据集DOTA上进行了实验，旋转框的目标检测网络的平均精度均值达到了68.5%，检测速度达到了每秒17.4帧图像。

Abstract

With the rapid development of deep learning

more and more target detection algorithms based on anchor frame are applied to remote sensing images in recent years.However

the cost of improving the accuracy of the algorithm is to sacrifice the detection speed.Therefore

the target detection network framework of anchor free was chosen

and a remote sensing detection algorithm of rotating frame was proposed according to the characteristics of remote sensing scene.A simple and effective representation of rotating frame was proposed according to the spatial position relationship between rotating frame and its external rectangular frame.In addition

an angle sensitive attention mechanism was designed to assist the detection of rotating targets.By introducing angle information

the detection ability of the model for rotating targets was improved.The proposed algorithm was tested on the open remote sensing dataset DOTA.The mean average precision of target detection network of rotating frame is 68.5% and the detection speed is 17.4 frames per second.

关键词

Keywords

references

GIRSHICK R , DONAHUE J , DARRELL T , et al . Rich feature hierarchies for accurate object detection and semantic segmentation [C ] // Proceedings of the 2014 IEEE Conference on Computer Vision and Pattern Recognition . Piscataway:IEEE Press , 2014 : 580 - 587 .

GIRSHICK R , . Fast R-CNN [C ] // Proceedings of the 2015 IEEE International Conference on Computer Vision . Piscataway:IEEE Press , 2015 : 1440 - 1448 .

REN S Q , HE K M , GIRSHICK R , et al . Faster R-CNN:towards real-time object detection with region proposal networks [C ] // Proceedings of the IEEE Transactions on Pattern Analysis and Machine Intelligence . Piscataway:IEEE Press , 2015 : 1137 - 1149 .

HE K M , GKIOXARI G , DOLLÁR P ,, et al . Mask R-CNN [C ] // Proceedings of the 2017 IEEE International Conference on Computer Vision . Piscataway:IEEE Press , 2017 : 2980 - 2988 .

HUANG L C , YANG Y , DENG Y F , et al . DenseBox:unifying landmark localization with end to end object detection [C ] // Proceedings of the IEEE International Conference on Computer Vision . Piscataway:IEEE Press , 2015 .

REDMON J , DIVVALA S , GIRSHICK R , et al . You only look once:unified,real-time object detection [C ] // Proceedings of the 2016 IEEE Conference on Computer Vision and Pattern Recognition . Piscataway:IEEE Press , 2016 : 779 - 788 .

LAW H , DENG J . CornerNet:detecting objects as paired keypoints [J ] . International Journal of Computer Vision , 2020 , 128 ( 3 ): 642 - 656 .

LIN T Y , MAIRE M , BELONGIE S , et al . Microsoft COCO:common objects in context [M ] // Computer Vision-ECCV 2014 . Cham : Springer International Publishing , 2014 : 740 - 755 .

ZHOU X Y , WANG D Q , KRÄHENBÜHL P , . Objects as points [J ] . arXiv preprint,2019,arXiv:1904.07850 .

ZHOU X Y , ZHUO J C , KRAHENBUHL P . Bottom-up object detection by grouping extreme and center points [C ] // Proceedings of the 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition . Piscataway:IEEE Press , 2019 .

MA T S , TIAN W H , KUANG P , et al . An anchor-free object detector with novel corner matching method [J ] . Knowledge-Based Systems , 2021 , 224 : 107083 .

LI Q M , QIANG H , LI J . Conditional random fields as message passing mechanism in anchor-free network for multi-scale pedestrian detection [J ] . Information Sciences , 2021 , 550 : 1 - 12 .

CHENG G , ZHOU P C , HAN J W . Learning rotation-invariant convolutional neural networks for object detection in VHR optical remote sensing images [J ] . IEEE Transactions on Geoscience and Remote Sensing , 2016 , 54 ( 12 ): 7405 - 7415 .

HE K M , ZHANG X Y , REN S Q , et al . Deep residual learning for image recognition [C ] // Proceedings of the 2016 IEEE Conference on Computer Vision and Pattern Recognition . Piscataway:IEEE Press , 2016 : 770 - 778 .

LIN T Y , DOLLÁR P ,, GIRSHICK R , et al . Feature pyramid networks for object detection [C ] // Proceedings of the 2017 IEEE Conference on Computer Vision and Pattern Recognition . Piscataway:IEEE Press , 2017 : 936 - 944 .

HU J , SHEN L , SUN G . Squeeze-and-excitation networks [C ] // Proceedings of the 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition . Piscataway:IEEE Press , 2018 : 7132 - 7141 .

CUBUK E D , ZOPH B , SHLENS J , et al . RandAugment:practical automated data augmentation with a reduced search space [C ] // Proceedings of the 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops . Piscataway:IEEE Press , 2020 : 3008 - 3017 .

MA J Q , SHAO W Y , YE H , et al . Arbitrary-oriented scene text detection via rotation proposals [J ] . IEEE Transactions on Multimedia , 2018 , 20 ( 11 ): 3111 - 3122 .

YANG X , LIU Q Q , YAN J C , et al . R3det:refined single-stage detector with feature refinement for rotating object [J ] . arXiv preprint,2019,arXiv:1908.05612 .

JIANG Y Y , ZHU X Y , WANG X B , et al . R2CNN:rotational region CNN for orientation robust scene text detection [J ] . arXiv preprint,2017,arXiv:1706.09579 .

浏览量

608

下载量

CSCD

文章被引用时，请邮件提醒。

提交

工具集

关联资源

基于改进YOLOv5s的小人脸检测

基于稠密块和注意力机制的肺部病理图像异常细胞分割

基于情感信息融合注意力机制的抑郁症识别

基于差分特征注意力机制的无锚框多光谱行人检测算法

基于改进 YOLOv5s 的小人脸检测