Opto-Electronic Engineering, Volume. 50, Issue 12, 230218-1(2024)

Improved CSTrack algorithm for multi-class ship multi-object tracking

Zhian Yuan, Yu Gu*, and Gan Ma
Author Affiliations
  • School of Automation, Hangzhou Dianzi University, Hangzhou, Zhejiang 310018, China
  • show less

    Due to the difficulties of complex backgrounds and large-scale differences between objects during the process of ship multi-object tracking in sea-surface scenarios, an improved CSTrack algorithm for ship multi-object tracking is proposed in this paper. Firstly, as violent decoupling is used in the CSTrack algorithm to decompose neck features and cause object feature loss, an improved cross-correlation decoupling network that combines the Res2net module (RES_CCN) is proposed, and thus more fine-grained features can be obtained. Secondly, to improve the tracking performance of multi-class ships, the decoupled design of the detection head network is used to predict the class, confidence, and position of objects, respectively. Finally, the MOT2016 dataset is used for the ablation experiment to verify the effectiveness of the proposed module. When tested on the Singapore maritime dataset, the multiple object tracking accuracy of the proposed algorithm is improved by 8.4% and the identification F1 score is increased by 3.1%, which are better than those of the ByteTrack and other algorithms. The proposed algorithm has the advantages of high tracking accuracy and low error detection rate and is suitable for ship multi-object tracking in sea-surface scenarios.

    Tools

    Get Citation

    Copy Citation Text

    Zhian Yuan, Yu Gu, Gan Ma. Improved CSTrack algorithm for multi-class ship multi-object tracking[J]. Opto-Electronic Engineering, 2024, 50(12): 230218-1

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Research Articles

    Received: Sep. 1, 2023

    Accepted: Dec. 5, 2023

    Published Online: Mar. 26, 2024

    The Author Email: Gu Yu (谷雨)

    DOI:10.12086/oee.2023.230218

    Topics