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

被引:13
|
作者
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 条
  • [31] SW-YOLOX: A YOLOX-based real-time pedestrian detector with shift window-mixed attention mechanism
    Tsai, Chi-Yi
    Wang, Run-Yu
    Chiu, Yu-Chen
    NEUROCOMPUTING, 2024, 606
  • [32] DARDet: A Dense Anchor-Free Rotated Object Detector in Aerial Images
    Zhang, Feng
    Wang, Xueying
    Zhou, Shilin
    Wang, Yingqian
    IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2022, 19
  • [33] Pneumonia detection based on RSNA dataset and anchor-free deep learning detector
    Wu, Linghua
    Zhang, Jing
    Wang, Yilin
    Ding, Rong
    Cao, Yueqin
    Liu, Guiqin
    Liufu, Changsheng
    Xie, Baowei
    Kang, Shanping
    Liu, Rui
    Li, Wenle
    Guan, Furen
    SCIENTIFIC REPORTS, 2024, 14 (01)
  • [34] HAFREE: A Heatmap-Based Anchor-Free Detector for Apple Defect Detection
    Bui Ngoc Han, Nguyen
    Lee, Ju-Hwan
    Thanh Vu, Dang
    Murtza, Iqbal
    Kim, Hyoung-Gook
    Kim, Jin-Young
    IEEE ACCESS, 2024, 12 : 182799 - 182813
  • [35] CPS-Det: An Anchor-Free Based Rotation Detector for Ship Detection
    Yang, Yi
    Pan, Zongxu
    Hu, Yuxin
    Ding, Chibiao
    REMOTE SENSING, 2021, 13 (11)
  • [36] Pneumonia detection based on RSNA dataset and anchor-free deep learning detector
    Linghua Wu
    Jing Zhang
    Yilin Wang
    Rong Ding
    Yueqin Cao
    Guiqin Liu
    Changsheng Liufu
    Baowei Xie
    Shanping Kang
    Rui Liu
    Wenle Li
    Furen Guan
    Scientific Reports, 14
  • [37] Anchor-free Object Detection Algorithm Based on Double Branch Feature Fusion
    Hou Zhiqiang
    Guo Hao
    Ma Sugang
    Cheng Huanhuan
    Bai Yu
    Fan Jiulun
    JOURNAL OF ELECTRONICS & INFORMATION TECHNOLOGY, 2022, 44 (06) : 2175 - 2183
  • [38] Anchor-free object detection network based on non-local operation
    Wu, Peng
    Wang, Chen
    Wang, Junxiao
    Wang, Yuanzhi
    MULTIMEDIA TOOLS AND APPLICATIONS, 2023, 83 (19) : 56249 - 56259
  • [39] Using Anchor-Free Object Detectors to Detect Surface Defects
    Liu, Jiaxue
    Zhang, Chao
    Li, Jianjun
    PROCESSES, 2024, 12 (12)
  • [40] GEOSPATIAL OBJECT DETECTION WITH SINGLE SHOT ANCHOR-FREE NETWORK
    Guo, Yiyou
    Ji, Jinsheng
    Lu, Xiankai
    Xie, Huan
    Tong, Xiaohua
    IGARSS 2020 - 2020 IEEE INTERNATIONAL GEOSCIENCE AND REMOTE SENSING SYMPOSIUM, 2020, : 280 - 283