Open Access iconOpen Access

ARTICLE

crossmark

A Novel YOLOv5s-Based Lightweight Model for Detecting Fish’s Unhealthy States in Aquaculture

Bing Shi1,*, Jianhua Zhao1, Bin Ma1, Juan Huan2, Yueping Sun3

1 School of Mechanical Engineering and Rail Transit, Changzhou University, Changzhou, 213164, China
2 School of Microelectronics and Control Engineering, Changzhou University, Changzhou, 213164, China
3 School of Electrical and Information Engineering, Jiangsu University, Zhenjiang, 212013, China

* Corresponding Author: Bing Shi. Email: email

Computers, Materials & Continua 2024, 81(2), 2437-2456. https://doi.org/10.32604/cmc.2024.056377

Abstract

Real-time detection of unhealthy fish remains a significant challenge in intensive recirculating aquaculture. Early recognition of unhealthy fish and the implementation of appropriate treatment measures are crucial for preventing the spread of diseases and minimizing economic losses. To address this issue, an improved algorithm based on the You Only Look Once v5s (YOLOv5s) lightweight model has been proposed. This enhanced model incorporates a faster lightweight structure and a new Convolutional Block Attention Module (CBAM) to achieve high recognition accuracy. Furthermore, the model introduces the α-SIoU loss function, which combines the α-Intersection over Union (α-IoU) and Shape Intersection over Union (SIoU) loss functions, thereby improving the accuracy of bounding box regression and object recognition. The average precision of the improved model reaches 94.2% for detecting unhealthy fish, representing increases of 11.3%, 9.9%, 9.7%, 2.5%, and 2.1% compared to YOLOv3-tiny, YOLOv4, YOLOv5s, GhostNet-YOLOv5, and YOLOv7, respectively. Additionally, the improved model positively impacts hardware efficiency, reducing requirements for memory size by 59.0%, 67.0%, 63.0%, 44.7%, and 55.6% in comparison to the five models mentioned above. The experimental results underscore the effectiveness of these approaches in addressing the challenges associated with fish health detection, and highlighting their significant practical implications and broad application prospects.

Keywords


Cite This Article

APA Style
Shi, B., Zhao, J., Ma, B., Huan, J., Sun, Y. (2024). A novel yolov5s-based lightweight model for detecting fish’s unhealthy states in aquaculture. Computers, Materials & Continua, 81(2), 2437-2456. https://doi.org/10.32604/cmc.2024.056377
Vancouver Style
Shi B, Zhao J, Ma B, Huan J, Sun Y. A novel yolov5s-based lightweight model for detecting fish’s unhealthy states in aquaculture. Comput Mater Contin. 2024;81(2):2437-2456 https://doi.org/10.32604/cmc.2024.056377
IEEE Style
B. Shi, J. Zhao, B. Ma, J. Huan, and Y. Sun, “A Novel YOLOv5s-Based Lightweight Model for Detecting Fish’s Unhealthy States in Aquaculture,” Comput. Mater. Contin., vol. 81, no. 2, pp. 2437-2456, 2024. https://doi.org/10.32604/cmc.2024.056377



cc Copyright © 2024 The Author(s). Published by Tech Science Press.
This work is licensed under a Creative Commons Attribution 4.0 International License , which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.
  • 368

    View

  • 142

    Download

  • 0

    Like

Share Link