|
--- |
|
license: gpl-3.0 |
|
--- |
|
# 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](https://huggingface.co/datasets/jero98772/CuraPeces_Background) |
|
[better dataset(removed background)](https://huggingface.co/datasets/jero98772/CuraPeces_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! ππ |