HD-YOLO: The hydroid of aurelia detector

Weihong Bi; Xinyuan Sun; Jiaxin Li; Yun Jin

doi:10.1016/j.marpolbul.2024.117346

HD-YOLO: The hydroid of aurelia detector

Mar Pollut Bull. 2024 Nov 29:210:117346. doi: 10.1016/j.marpolbul.2024.117346. Online ahead of print.

Authors

Weihong Bi¹, Xinyuan Sun², Jiaxin Li², Yun Jin³

Affiliations

¹ School of information Science and Engineering, Yanshan University, Qinhuangdao 066004, China. Electronic address: bwhong@ysu.edu.cn.
² School of information Science and Engineering, Yanshan University, Qinhuangdao 066004, China.
³ Department of Mechanical and Electrical Engineering, Hebei Construction Material Vocational and Technical College, Qinhuangdao 066004, China.

PMID: 39615339
DOI: 10.1016/j.marpolbul.2024.117346

Abstract

In recent years, jellyfish outbreaks have been frequent in near-shore waters and have become a worldwide problem. At present, the main species that form jellyfish disasters along the coast of China is aurelia. In order to control jellyfish outbreaks from the source of growth, this paper proposed a hydroid detector HD-YOLO improved from YOLOv7, and established a hydroid of aurelia dataset. Firstly, the original dataset was established by acquiring images of hydroids via the web, in the laboratory and undersea photography. Secondly, MSRCR, a multi-scale enhancement algorithm with color recovery, was used to improve the underwater image quality. Then the dataset is augmented by the combination of Mosaic and Mixup. Finally, the effectiveness of HD-YOLO, underwater image enhancement and dataset augmentation methods were verified through a series of comparative experiments. This study provides a more accurate and faster method for the hydroid of aurelia detection and a more generalised dataset.

Keywords: Hydroid of aurelia; MSRCR; Underwater object detection; YOLOv7.