Sw-YoloX: An anchor-free detector based transformer for sea surface object detection

被引:7
|
作者
Ding, Jiangang [1 ]
Li, Wei [1 ]
Pei, Lili [1 ]
Yang, Ming [1 ]
Ye, Chao [1 ]
Yuan, Bo [1 ]
机构
[1] Changan Univ, Sch Informat Engn, Xian 710064, Shaanxi, Peoples R China
基金
中国国家自然科学基金;
关键词
Sea surface object detection; Sw-YoloX; Transformer; YoloX; Self-training classifier;
D O I
10.1016/j.eswa.2023.119560
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
To cope with the challenge of blurred images of sea surface objects caused by the complex and undulating sea surface environment, we propose Sw-YoloX, which can utilize the global modeling ability to encode the key semantics of sea surface objects, thereby obtaining global features that cannot be captured by CNN. Then the convolutional block attention module (CBAM) and atrous spatial pyramid pooling (ASPP) module are integrated in the neck of the detector, and the decoupled head is used as the prediction part. In addition, we also integrate multiple training strategies to effectively improve the detector performance, such as simple optimal transport assignment (SimOTA) strategy and multi-model integration. Finally, we construct the XM-10000 dataset for validation based on sea surface monitoring data in Xiamen, China. With end-to-end training, Sw-YoloX achieves higher performance than baseline and mainstream detector, with F1-Score is 78.1, mean average precision (mAP) is 54.4, and average recall (AR) is 72.0. This research, which has now been deployed in the coastal defense department in Xiamen, China, has important implications for searching for survivors and preventing smuggling.
引用
收藏
页数:11
相关论文
共 50 条
  • [1] R2YOLOX: A Lightweight Refined Anchor-Free Rotated Detector for Object Detection in Aerial Images
    Liu, Fei
    Chen, Renwen
    Zhang, Junyi
    Xing, Kailing
    Liu, Hao
    Qin, Jinchang
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2022, 60
  • [2] A fully convolutional anchor-free object detector
    Taoshan Zhang
    Zheng Li
    Zhikuan Sun
    Lin Zhu
    [J]. The Visual Computer, 2023, 39 : 569 - 580
  • [3] A fully convolutional anchor-free object detector
    Zhang, Taoshan
    Li, Zheng
    Sun, Zhikuan
    Zhu, Lin
    [J]. VISUAL COMPUTER, 2023, 39 (02): : 569 - 580
  • [4] MAOD: An Efficient Anchor-Free Object Detector Based on MobileDet
    Chen, Dong
    Shen, Hao
    [J]. IEEE ACCESS, 2020, 8 : 86564 - 86572
  • [5] FCOSR: A Simple Anchor-Free Rotated Detector for Aerial Object Detection
    Li, Zhonghua
    Hou, Biao
    Wu, Zitong
    Ren, Bo
    Yang, Chen
    [J]. REMOTE SENSING, 2023, 15 (23)
  • [6] AN ORIENTATION-AWARE ANCHOR-FREE DETECTOR FOR AERIAL OBJECT DETECTION
    Duan, Mudi
    Meng, Ran
    Xiao, Liang
    [J]. 2022 IEEE INTERNATIONAL GEOSCIENCE AND REMOTE SENSING SYMPOSIUM (IGARSS 2022), 2022, : 3075 - 3078
  • [7] Gaussian Aware Anchor-Free Rotated Detector for Aerial Object Detection
    Zhang, Shuai
    Zhang, Cunyuan
    Yu, Lijian
    Ji, Wenyu
    Zhi, Xiyang
    [J]. IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2024, 21 : 1 - 5
  • [8] FCOS: A Simple and Strong Anchor-Free Object Detector
    Tian, Zhi
    Shen, Chunhua
    Chen, Hao
    He, Tong
    [J]. IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2022, 44 (04) : 1922 - 1933
  • [9] ElDet: An Anchor-Free General Ellipse Object Detector
    Wang, Tianhao
    Lu, Changsheng
    Shao, Ming
    Yuan, Xiaohui
    Xia, Siyu
    [J]. COMPUTER VISION - ACCV 2022, PT III, 2023, 13843 : 223 - 238
  • [10] TomatoDet: Anchor-free detector for tomato detection
    Liu, Guoxu
    Hou, Zengtian
    Liu, Hongtao
    Liu, Jun
    Zhao, Wenjie
    Li, Kun
    [J]. FRONTIERS IN PLANT SCIENCE, 2022, 13