Computer Science ›› 2024, Vol. 51 ›› Issue (6A): 230700117-7.doi: 10.11896/jsjkx.230700117

• Image Processing & Multimedia Technolog • Previous Articles     Next Articles

Ship Detection and Recognition of Optical Remote Sensing Images for Embedded Platform

HE Xinyu, LU Chenxin, FENG Shuyi, OUYANG Shangrong, MU Wentao   

  1. Shanghai Aerospace Electronic Technology Research Institute,Shanghai 201109,China
  • Published:2024-06-06
  • About author:HE Xinyu,born in 1999,postgraduate.His main research interest is image processing applied to satellite.
    FENG Shuyi,born in 1984,professor.His main research interests include image processing and artificial intelligence.

Abstract: The construction of a maritime power is a current strategic direction for China’s vigorous development.In response to the low detection and classification recognition rate and slow operation speed of existing deep learning-based remote sensing image ship target detection and classification algorithms on embedded platforms,this paper proposes an improved Mix-YOLO network model based on the Cambricon-MLU220 embedded platform.The model is based on the YOLOv7-tiny network as the basic framework.Firstly,the MobileNet series network module is introduced to replace the feature extraction network partially,reducing the network parameter volume.Then,the ULSAM attention mechanism is introduced to enhance the network’s learning and classification ability,reducing the false alarm rate.Finally,in order to make the detection speed improvement effect more obvious on the embedded platform,the network model is programmed by splitting the large module into small modules.Experimental results show that the Mix-YOLO algorithm reduces the parameter volume and calculation by 39.70% and 29.70%,respectively,on the basis of the original YOLOv7-tiny network.The processing frame rate is increased from 97.27 fps to 120.88 fps,and the accuracy is improved by 7.7%.It can achieve real-time detection and recognition of ship targets in remote sensing images.

Key words: Optical remote sensing image, Ship detection, Lightweight network, Attention mechanism, Embedded platform

CLC Number: 

  • TP751
[1]GIRSHICK R.Fast R-CNN[C]//2015 IEEE International Conference on Computer Vision(ICCV).Santiago,Chile.IEEE,2016:1440-1448.
[2]JIANG H Z,LEARNED-MILLER E.Face Detection with the Faster R-CNN[C]//2017 12th IEEE International Conference on Automatic Face & Gesture Recognition(FG 2017).Washington,DC,USA.IEEE,2017:650-657.
[3]HE K M,GKIOXARI G,DOLLÁR P,et al.Mask R-CNN[C]//2017 IEEE International Conference on Computer Vision(ICCV).Venice,Italy,IEEE,2017:2980-2988.
[4]LIU W,ANGUELOV D,ERHAN D,et al.SSD:single shotMultiBox detector[J].Computer Vision,2016,1(14):21-37.
[5]WANG Y Y,WANG C,ZHANG H,et al.Automatic ship detection based on RetinaNet using multi-resolution Gaofen-3 imagery[J].Remote Sensing,2019,11(5):531.
[6]REDMON J,DIVVALA S,GIRSHICK R,et al.You Only Look Once:Unified,Real-Time Object Detection[C]//2016 IEEE Conference on Computer Vision and Pattern Recognition(CVPR).Las Vegas,NV,USA,2016:779-788.
[7]REDMON J,FARHADI A.YOLO9000:Better,Faster,Stronger[C]//IEEE Conference on Computer Vision & Pattern Recognition.IEEE,2017:6517-6525.
[8]REDMON J,FARHADI A.Yolov3:An incre-mental improve-ment[J].arXiv:1804.02767,2018.
[9]BOCHKOVSKIY A,WANG C Y,LIAO H Y M.YOLOv4:optimal speed and accuracy of object detection[J].arXiv:2004.10934,2020.
[10]WANG C Y,BOCHKOVSKIY A,LIAO H Y M.YOLOv7:Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors[C]//CVPR.2023:7464-7475.
[11]CUI J H,ZHANG Y Z,WANG Z,et al.Light-weight object detection networks for embedded platform[J].Acta Optica Sinica,2019,39(4):0415006.
[12]SU N,HUANG Z,YAN Y,et al.Detect Larger at Once:Large-Area Remote-Sensing Image Arbitrary-Oriented Ship Detection[J].IEEE Geoscience and Remote Sensing Letters,2022,19:1-5.
[13]YU J,PENG X,LI S,et al.A Lightweight Ship Detection Me-thod in Optical Remote Sensing Image under Cloud Interference[C]//2021 IEEE International Instrumentation and Measurement Technology Conference(I2MTC).IEEE,2021.
[14]NONG Y J,WANG J J.Remote sensing image caption method based on attention and reinforcement learning[J].Acta Optica Sinica,2021,41(22):2228001.
[15]LIU T,ZHOU B J,ZHAO Y S,et al.Ship Detection Algorithm based on Improved YOLO V5[C]//2021 6th International Conference on Automation,Control and Robotics Engineering(CACRE).Dalian,China,2021:483-487.
[16]WANG H Y,WANG C P,FU Q,et al.Lightweight Ship Detection Based on Optical Remote Sensing Images for Embedded Platform[J].Acta Optica Sinica,2023,43(12):1212001.
[17]LIU X W,PIAO Y J,ZHENG L L,et al.Ship detection for complex scene images of space optical remote sensing[J].Optics and Precision Engineering,2023,31(6):892-904.
[18]HOWARD A,SANDLER M,CHU G,et al.Searching for MobileNetV3[C]//Proceeding of the IEEE/CVF International Conference on Computer Vision.2019:1314-1324.
[19]JIE H,LI S,GANG S.Squeeze-and-Excitation Networks[C]//2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition(CVPR).IEEE,2018.
[20]SAINI R,JHA N K,DAS B,et al.ULSAM:Ultra-Lightweight Subspace Attention Module for Compact Convolutional Neural Networks[C]//Proceedings of the IEEE/CVF Winter Conference on Applications of Computer Vision.2020:1627-1636.
[1] ZHANG Le, YU Ying, GE Hao. Mural Inpainting Based on Fast Fourier Convolution and Feature Pruning Coordinate Attention [J]. Computer Science, 2024, 51(6A): 230400083-9.
[2] SUN Yang, DING Jianwei, ZHANG Qi, WEI Huiwen, TIAN Bowen. Study on Super-resolution Image Reconstruction Using Residual Feature Aggregation NetworkBased on Attention Mechanism [J]. Computer Science, 2024, 51(6A): 230600039-6.
[3] QUE Yue, GAN Menghan, LIU Zhiwei. Object Detection with Receptive Field Expansion and Multi-branch Aggregation [J]. Computer Science, 2024, 51(6A): 230600151-6.
[4] ZHAI Yunkai, QIAO Zhengwen, QIAO Yan. Forecasting Teleconsultation Demand Based on LSTM and Attention Mechanism [J]. Computer Science, 2024, 51(6A): 230800119-7.
[5] ZHANG Lanxin, XIANG Ling, LI Xianze, CHEN Jinpeng. Intelligent Fault Diagnosis Method for Rolling Bearing Based on SAMNV3 [J]. Computer Science, 2024, 51(6A): 230700167-6.
[6] ZHU Yuliang, LIU Juntao, RAO Ziyun, ZHANG Yi, CAO Wanhua. Knowledge Reasoning Model Combining HousE with Attention Mechanism [J]. Computer Science, 2024, 51(6A): 230600209-8.
[7] BAI Yu, WANG Xinzhe. Study on Hypernymy Recognition Based on Combined Training of Attention Mechanism and Prompt Learning [J]. Computer Science, 2024, 51(6A): 230700226-5.
[8] WANG Guogang, DONG Zhihao. Lightweight Image Semantic Segmentation Based on Attention Mechanism and Densely AdjacentPrediction [J]. Computer Science, 2024, 51(6A): 230300204-8.
[9] LIU Xiaohu, CHEN Defu, LI Jun, ZHOU Xuwen, HU Shan, ZHOU Hao. Speaker Verification Network Based on Multi-scale Convolutional Encoder [J]. Computer Science, 2024, 51(6A): 230700083-6.
[10] WU Chunming, WANG Tiaojun. Study on Defect Detection Algorithm of Transmission Line in Complex Background [J]. Computer Science, 2024, 51(6A): 230500178-6.
[11] LYU Yiming, WANG Jiyang. Iron Ore Image Classification Method Based on Improved Efficientnetv2 [J]. Computer Science, 2024, 51(6A): 230600212-6.
[12] WU Chunming, LIU Yali. Method for Lung Nodule Detection on CT Images Using Improved YOLOv5 [J]. Computer Science, 2024, 51(6A): 230500019-6.
[13] XIAO Yahui, ZHANG Zili, HU Xinrong, PENG Tao, ZHANG Jun. Clothing Image Segmentation Method Based on Deeplabv3+ Fused with Attention Mechanism [J]. Computer Science, 2024, 51(6A): 230900153-7.
[14] LANG Lang, CHEN Xiaoqin, LIU Sha, ZHOU Qiang. Detection of Pitting Defects on the Surface of Ball Screw Drive Based on Improved Deeplabv3+ Algorithm [J]. Computer Science, 2024, 51(6A): 240200058-6.
[15] QIAO Hong, XING Hongjie. Attention-based Multi-scale Distillation Anomaly Detection [J]. Computer Science, 2024, 51(6A): 230300223-11.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!