Curapeces

Fish Disease Classification Model

This repository contains a model trained for classifying various fish diseases using Hugging Face's Transformers library. The model is trained to classify the following fish diseases:

1. Tumor and deformity attack
2. Polydactyly
3. Hydrocephalus and swim bladder
4. Holes in the head
5. Nopez (I couldn't find a direct translation, it seems like a term specific to a certain context)
6. Parasite in the mouth
7. Catfish burn
8. Gills
9. Lernea worm
10. Fungi
11. Ich white spot
12. Blind eye
13. Fin rot
14. Healthy

Dataset

The model is trained on a diverse dataset containing images of fish affected by different diseases as well as healthy fish samples. The dataset is labeled with the aforementioned disease categories to facilitate supervised training.

dataset base better dataset(removed background)

Model Architecture

The classification model is built upon state-of-the-art deep learning architecture, leveraging the power of convolutional neural networks (CNNs) for image classification tasks. It utilizes transfer learning techniques, starting with a pre-trained backbone network (e.g., ResNet, EfficientNet) and fine-tuning it on the fish disease dataset to adapt to the specific classification task.

Happy classifying fish diseases! 🐟🔍