An Improved Deep Learning Model for Underwater Species Recognition in Aquaculture

The ability to differentiate between various fish species plays an essential role in aquaculture. It helps to protect their populations and monitor their health situations and their nutrient systems. However, old machine learning methods are unable to detect objects in images with complex background...

Full description

Bibliographic Details
Main Authors: Mahdi Hamzaoui, Mohamed Ould-Elhassen Aoueileyine, Lamia Romdhani, Ridha Bouallegue
Format: Article
Language:English
Published: MDPI AG 2023-10-01
Series:Fishes
Subjects:
Online Access:https://www.mdpi.com/2410-3888/8/10/514
Description
Summary:The ability to differentiate between various fish species plays an essential role in aquaculture. It helps to protect their populations and monitor their health situations and their nutrient systems. However, old machine learning methods are unable to detect objects in images with complex backgrounds and especially in low-light conditions. This paper aims to improve the performance of a YOLO v5 model for fish recognition and classification. In the context of transfer learning, our improved model FishDETECT uses the pre-trained FishMask model. Then it is tested in various complex scenes. The experimental results show that FishDETECT is more effective than a simple YOLO v5 model. Using the evaluation metrics Precision, Recall, and mAP50, our new model achieved accuracy rates of 0.962, 0.978, and 0.995, respectively.
ISSN:2410-3888