panda992/fish_disease_datasets
🐟 Fish Disease Classifier (ViT)
This model is a fine-tuned version of google/vit-base-patch16-224-in21k, trained on a custom fish disease image dataset for Indian aquaculture. ✅ Detected Classes (Fish)
Bacterial Red Disease
Bacterial diseases – Aeromoniasis
Bacterial Gill Disease
Fungal diseases (Saprolegniasis)
Parasitic diseases
Viral diseases (White Tail Disease)
Healthy Fish
⚠️ Planned Prawn Model (Upcoming)
We are currently working on a separate fine-tuned model to detect:
Bacterial Gill Disease (BG)
White Spot Syndrome Virus (WSSV)
Healthy Prawn
This model will be released in the next version once prawn dataset collection and training is complete. 📊 Evaluation Metrics Metric Value Accuracy 97.28% Validation Loss 0.0866 Final Epoch 4 🧠 Model Description
Architecture: Vision Transformer (ViT)
Base model: google/vit-base-patch16-224-in21k
Dataset: Custom-labeled images of freshwater fish diseases
Data augmentation: Albumentations
Optimized for WhatsApp-based diagnosis tools
🚜 Intended Use
This model is optimized for:
Farmers needing fast disease detection via image
WhatsApp or mobile-based advisory tools
NGO/hatchery/government pilots in India and South Asia
🏋️ Training Summary
Learning rate: 0.0002
Batch size: 16 (train) / 8 (eval)
Epochs: 4
Mixed Precision: AMP
Framework: Hugging Face Transformers, PyTorch