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 条
  • [11] Coordinate-based anchor-free module for object detection
    Zhiyong Tang
    Jianbing Yang
    Zhongcai Pei
    Xiao Song
    Applied Intelligence, 2021, 51 : 9066 - 9080
  • [12] Feature Alignment in Anchor-Free Object Detection
    Gao, Feng
    Cai, Yeyun
    Deng, Fang
    Yu, Chengpu
    Chen, Jie
    IEEE TRANSACTIONS ON CIRCUITS AND SYSTEMS FOR VIDEO TECHNOLOGY, 2023, 33 (08) : 3799 - 3810
  • [13] Coordinate-based anchor-free module for object detection
    Tang, Zhiyong
    Yang, Jianbing
    Pei, Zhongcai
    Song, Xiao
    APPLIED INTELLIGENCE, 2021, 51 (12) : 9066 - 9080
  • [14] An Anchor-Free Lightweight Object Detection Network
    Wang, Weina
    Gou, Yunyan
    IEEE ACCESS, 2023, 11 : 110361 - 110374
  • [15] Anchor-free object detection with mask attention
    Yang, He
    Fan, Beibei
    Guo, Lingling
    EURASIP JOURNAL ON IMAGE AND VIDEO PROCESSING, 2020, 2020 (01)
  • [16] Anchor-free object detection with mask attention
    He Yang
    Beibei Fan
    Lingling Guo
    EURASIP Journal on Image and Video Processing, 2020
  • [17] An Anchor-Free Detector With Channel-Based Prior and Bottom-Enhancement for Underwater Object Detection
    Ouyang, Wenjia
    Wei, Yanhui
    IEEE SENSORS JOURNAL, 2023, 23 (20) : 24800 - 24811
  • [18] An anchor-free object detector with novel corner matching method
    Ma, Tingsong
    Tian, Wenhong
    Kuang, Ping
    Xie, Yuanlun
    KNOWLEDGE-BASED SYSTEMS, 2021, 224
  • [19] MEAD: a Mask-guidEd Anchor-free Detector for oriented aerial object detection
    Zewen He
    Zhida Ren
    Xuebing Yang
    Yang Yang
    Wensheng Zhang
    Applied Intelligence, 2022, 52 : 4382 - 4397
  • [20] MEAD: a Mask-guidEd Anchor-free Detector for oriented aerial object detection
    He, Zewen
    Ren, Zhida
    Yang, Xuebing
    Yang, Yang
    Zhang, Wensheng
    APPLIED INTELLIGENCE, 2022, 52 (04) : 4382 - 4397