EC-YOLOX: A Deep-Learning Algorithm for Floating Objects Detection in Ground Images of Complex Water Environments

Authors

Jiaxin He, Nanjing University
Yong Cheng, Nanjing University
Wei Wang, Nanjing University
Yakang Gu, Nanjing University
Yixuan Wang, Nanjing University
Wenjie Zhang, Nanjing University
Achyut Shankar, University of Warwick
Shitharth Selvarajan,, Leeds Beckett University
Sathish Kumar, Cleveland State UniversityFollow

Document Type

Article

Publication Date

2024

Publication Title

IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing

Abstract

Correct detection of floating objects in complex water environments is a challenge because of the problems of obscuration and dense floating objects. In view of the above issues, this article proposed a network called EC-YOLOX by introducing the coordinate attention (CA) and efficient channel attention (ECA) mechanism and improving the loss function to further the multifeature extraction and detection accuracy of floating objects. In this article, ablation experiments and comparison experiments were conducted on the river floating objects dataset. The ablation experiments showed that the ECA and CA mechanism played a great role in EC-YOLOX, which can reduce the missed detection rate by 5.86% and increase the mean average precision (mAP) by 5.53% compared with YOLOX. The EC-YOLOX was also applicable to different types of floating objects; the mAP of the ball, plastic garbage, plastic bag, leaf, milk box, grass, and branches were, respectively, improved by 4%, 4%, 4%, 6%, 4%, 18%, and 5%. The mAP of the comparison experiments was improved by 15.13%, 9.30%, and 8.03% compared to faster R-CNN, YOLOv5, and YOLOv3, respectively. This method facilitates the precise extraction of floating objects from images, which holds paramount importance for monitoring and safeguarding water environments. It offers significant contributions to water environment monitoring and protection.

DOI

10.1109/JSTARS.2024.3367713

Recommended Citation

He, Jiaxin; Cheng, Yong; Wang, Wei; Gu, Yakang; Wang, Yixuan; Zhang, Wenjie; Shankar, Achyut; Selvarajan,, Shitharth; and Kumar, Sathish, "EC-YOLOX: A Deep-Learning Algorithm for Floating Objects Detection in Ground Images of Complex Water Environments" (2024). Computer Science Faculty Publications. 7.
https://engagedscholarship.csuohio.edu/encs_facpub/7

Volume

17