Update README.md
Browse files
README.md
CHANGED
@@ -1,3 +1,38 @@
|
|
1 |
---
|
2 |
license: gpl-3.0
|
3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
---
|
2 |
license: gpl-3.0
|
3 |
---
|
4 |
+
# Fish Disease Classification Model
|
5 |
+
|
6 |
+
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:
|
7 |
+
|
8 |
+
|
9 |
+
1. **Tumor and deformity attack**
|
10 |
+
2. **Polydactyly**
|
11 |
+
3. **Hydrocephalus and swim bladder**
|
12 |
+
4. **Holes in the head**
|
13 |
+
5. **Nopez (I couldn't find a direct translation, it seems like a term specific to a certain context)**
|
14 |
+
6. **Parasite in the mouth**
|
15 |
+
7. **Catfish burn**
|
16 |
+
8. **Gills**
|
17 |
+
9. **Lernea worm**
|
18 |
+
10. **Fungi**
|
19 |
+
11. **Ich white spot**
|
20 |
+
12. **Blind eye**
|
21 |
+
13. **Fin rot**
|
22 |
+
14. **Healthy**
|
23 |
+
|
24 |
+
## Dataset
|
25 |
+
|
26 |
+
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.
|
27 |
+
|
28 |
+
[dataset base](https://huggingface.co/datasets/jero98772/CuraPeces_Background)
|
29 |
+
[better dataset(removed background)](https://huggingface.co/datasets/jero98772/CuraPeces_Removed_background)
|
30 |
+
|
31 |
+
## Model Architecture
|
32 |
+
|
33 |
+
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.
|
34 |
+
|
35 |
+
|
36 |
+
|
37 |
+
|
38 |
+
Happy classifying fish diseases! 🐟🔍
|