Online Multi-Object Tracking Method for Shrimps in High Density Using Multi-Task You Only Look Once Exceeding and Cascade Strategy
Yiran Liu,Beibei Li,Xinhui Zhou,Daoliang Li,Qingling Duan
DOI: https://doi.org/10.1016/j.engappai.2024.108882
IF: 8
2024-01-01
Engineering Applications of Artificial Intelligence
Abstract:Online multi-object tracking is essential for analyzing shrimp motion patterns and behavior characteristics, crucial for breeding and behavioristics research. Small, densely distributed shrimp often occlude each other, causing frequent identification switches during tracking. To address this, a multi-object tracking method for shrimps using a multi-task You Only Look Once eXceeding (YOLOX) and cascade strategy is proposed. In our method, a re-identification branch is attached to the YOLOX framework as a prediction head, achieving joint detection and embedding to provide tracking cues. Meanwhile, a global attention block is added to the detection branch to enhance the regression performance. Furthermore, an uncertainty loss function is employed to dynamically adjust the weights for detection and representation. A cascade strategy is implemented during tracking, considering appearance representations and objects with low scores, which resolves occlusion issues in dense scenes. The performance of the approach was verified using a video dataset of shrimps, achieving the best results compared to state-of-the-art methods with a Multi-Object Tracking Accuracy (MOTA) of 97.9%, a Multi-Object Tracking Precision (MOTP) of 0.16, and an F1-score of identification (IDF1) of 98.0%. The results indicate significant improvements over the baseline YOLOX model. The MOTA, IDF1, and Most Tracked increased by 4.2%, 2.0%, and 5.2% and the MOTP, identification switches, Fragments, and Most Lost decreased by 15.3%, 30.0%, 49.5%, 100%. These results demonstrate that this method can track shrimps online in dense scenarios, addressing issues of small targets and frequent identification switches, thus providing ideas for tracking other small objects in dense distributions.