Coralguard/Corals2
0
๐ชธ Coral Health Analysis System
An AI-powered system for coral species classification and health assessment using deep learning and color analysis.
๐ฏ Features
1. Coral Species Classification
- Uses EfficientNet-B0 CNN model
- Classifies into 4 coral types:
- ๐ชจ Boulder Coral
- ๐ฟ Branched Coral
- ๐ฝ๏ธ Plate Coral
- ๐ Soft Coral
- Displays confidence scores for each class
2. Health Assessment
- CoralWatch color-based health scoring
- Analyzes both dark and light color patches
- Provides health status:
- โ Healthy: Average score > 3
- โ ๏ธ Stressed/Bleached: Average score โค 3
3. Advanced Segmentation
- Removes water, sand, and background
- Isolates coral regions for accurate analysis
- Multi-strategy preprocessing
4. Visual Overlays
- Color-coded health visualization
- Segmentation masks
- Confidence charts
๐ How to Use
- Upload a coral image (JPG, PNG)
- Click "Analyze Coral"
- View results:
- Species classification
- Health status
- Color analysis
- Visual overlays
๐ฌ Technology Stack
- Deep Learning: PyTorch, EfficientNet-B0
- Computer Vision: OpenCV, scikit-learn
- Interface: Gradio
- Color Analysis: CoralWatch methodology
๐ Model Details
- Architecture: EfficientNet-B0
- Input Size: 224x224
- Training: ImageNet pretrained + fine-tuned on coral dataset
- Classes: 4 coral species
๐จ Health Scoring
Based on CoralWatch color reference chart:
- Score 1-2: Severely bleached (lightest)
- Score 3: Moderately stressed
- Score 4-6: Healthy (darkest)
๐ Citation
If you use this system in your research, please cite:
Coral Health Analysis System
Powered by EfficientNet-B0 and CoralWatch methodology๐ค Contributing
Contributions are welcome! Feel free to:
- Report issues
- Suggest improvements
- Add new features
๐ License
MIT License
Built with โค๏ธ for coral reef conservation
