Fish Classification System Using YOLOv3-ResNet18 Model for Mobile Phones

Suryadiputra Liawatimena,Edi Abdurachman,Agung Trisetyarso,Antoni Wibowo,Muhamad Keenan Ario,Ivan Sebastian Edbert
DOI: https://doi.org/10.21512/commit.v17i1.8107
2023-03-17
CommIT (Communication and Information Technology) Journal
Abstract:Every country in the world needs to report its fish production to the Food and Agriculture Organization of the United Nations (FAO) every year. In 2018, Indonesia ranked top five countries in fish production, with 8 million tons globally. Although it ranks top five, the fisheries in Indonesia are mostly dominated by traditional and small industries. Hence, a solution based on computer vision is needed to help detect and classify the fish caught every year. The research presents a method to detect and classify fish on mobile devices using the YOLOv3 model combined with ResNet18 as a backbone. For the experiment, the dataset used is four types of fish gathered from scraping across the Internet and taken from local markets and harbors with a total of 4,000 images. In comparison, two models are used: SSD-VGG and autogenerated model Huawei ExeML. The results show that the YOLOv3-ResNet18 model produces 98.45% accuracy in training and 98.15% in evaluation. The model is also tested on mobile devices and produces a speed of 2,115 ms on Huawei P40 and 3,571 ms on Realme 7. It can be concluded that the research presents a smaller-size model which is suitable for mobile devices while maintaining good accuracy and precision.
What problem does this paper attempt to address?