Update README.md
Browse files
README.md
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
|
|
| 1 |
title: Satellite Classification Dashboard
|
| 2 |
emoji: ๐ฐ๏ธ
|
| 3 |
colorFrom: blue
|
|
@@ -11,129 +12,165 @@ preload_from_hub:
|
|
| 11 |
- Bhavi23/MobilenetV2/multi_input_model_v1.keras
|
| 12 |
- Bhavi23/EfficientNet_B0/efficientnet_model.keras
|
| 13 |
- Bhavi23/DenseNet/densenet_model.keras
|
|
|
|
|
|
|
|
|
|
| 14 |
|
| 15 |
-
๐ฐ๏ธ Satellite Classification Dashboard
|
| 16 |
A comprehensive deep learning application for classifying satellite images using multiple state-of-the-art neural network models. This interactive dashboard allows you to compare different models, analyze their performance, and classify satellite images in real-time.
|
| 17 |
|
| 18 |
-
๐ Features
|
| 19 |
-
|
| 20 |
-
|
| 21 |
-
|
| 22 |
-
|
| 23 |
-
|
| 24 |
-
|
| 25 |
-
|
| 26 |
-
|
| 27 |
-
|
| 28 |
-
|
| 29 |
-
|
| 30 |
-
|
|
|
|
|
|
|
|
|
|
| 31 |
The system can classify 11 different types of satellites and space objects:
|
| 32 |
|
| 33 |
-
AcrimSat
|
| 34 |
-
Aquarius
|
| 35 |
-
Aura
|
| 36 |
-
Calipso
|
| 37 |
-
Cloudsat
|
| 38 |
-
CubeSat
|
| 39 |
-
Debris
|
| 40 |
-
Jason
|
| 41 |
-
Sentinel-6
|
| 42 |
-
TRMM
|
| 43 |
-
Terra
|
| 44 |
-
|
| 45 |
-
|
|
|
|
|
|
|
| 46 |
Visit the Hugging Face Space to use the application directly in your browser.
|
| 47 |
|
| 48 |
-
Local Installation
|
| 49 |
-
Clone the repository
|
|
|
|
| 50 |
git clone https://github.com/YOUR_USERNAME/satellite-classification-dashboard.git
|
| 51 |
cd satellite-classification-dashboard
|
| 52 |
-
|
|
|
|
|
|
|
|
|
|
| 53 |
pip install -r requirements.txt
|
| 54 |
-
|
|
|
|
|
|
|
|
|
|
| 55 |
streamlit run app.py
|
| 56 |
-
|
| 57 |
-
|
| 58 |
-
|
| 59 |
-
|
| 60 |
-
|
| 61 |
-
|
| 62 |
-
|
| 63 |
-
|
| 64 |
-
|
| 65 |
-
|
| 66 |
-
|
| 67 |
-
|
| 68 |
-
|
| 69 |
-
|
| 70 |
-
|
| 71 |
-
|
| 72 |
-
|
| 73 |
-
|
| 74 |
-
|
| 75 |
-
|
| 76 |
-
|
| 77 |
-
|
| 78 |
-
|
| 79 |
-
|
| 80 |
-
|
| 81 |
-
|
| 82 |
-
|
| 83 |
-
|
| 84 |
-
|
| 85 |
-
|
| 86 |
-
|
| 87 |
-
|
| 88 |
-
|
| 89 |
-
|
| 90 |
-
|
| 91 |
-
|
| 92 |
-
|
| 93 |
-
|
| 94 |
-
|
| 95 |
-
|
| 96 |
-
|
| 97 |
-
|
| 98 |
-
|
| 99 |
-
|
| 100 |
-
|
| 101 |
-
|
| 102 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 103 |
All models are hosted on Hugging Face Model Hub:
|
| 104 |
-
|
| 105 |
-
|
| 106 |
-
|
| 107 |
-
|
| 108 |
-
|
| 109 |
-
Data Pipeline
|
| 110 |
-
Image Upload: Supports PNG, JPG, JPEG formats
|
| 111 |
-
Preprocessing: Resize to 224x224, normalize to [0,1]
|
| 112 |
-
Model Inference: Multi-model prediction with timing
|
| 113 |
-
Post-processing: Confidence scoring and recommendations
|
| 114 |
-
|
| 115 |
-
|
| 116 |
-
|
| 117 |
-
Input
|
| 118 |
-
|
| 119 |
-
|
| 120 |
-
|
| 121 |
-
|
| 122 |
-
|
| 123 |
-
|
| 124 |
-
|
| 125 |
-
|
| 126 |
-
|
| 127 |
-
|
| 128 |
-
|
| 129 |
-
|
| 130 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 131 |
The application is deployed on Hugging Face Spaces using:
|
|
|
|
|
|
|
|
|
|
| 132 |
|
| 133 |
-
|
| 134 |
-
|
| 135 |
-
Resources: CPU-optimized for inference
|
| 136 |
-
Docker Deployment
|
| 137 |
FROM python:3.9-slim
|
| 138 |
|
| 139 |
WORKDIR /app
|
|
@@ -144,15 +181,19 @@ COPY . .
|
|
| 144 |
EXPOSE 7860
|
| 145 |
|
| 146 |
CMD ["streamlit", "run", "app.py", "--server.port=7860", "--server.address=0.0.0.0"]
|
| 147 |
-
|
|
|
|
|
|
|
| 148 |
We welcome contributions! Please follow these steps:
|
| 149 |
|
| 150 |
-
Fork the repository
|
| 151 |
-
Create a feature branch (git checkout -b feature/amazing-feature)
|
| 152 |
-
Commit changes (git commit -m 'Add amazing feature')
|
| 153 |
-
Push to branch (git push origin feature/amazing-feature)
|
| 154 |
-
Open a Pull Request
|
| 155 |
-
|
|
|
|
|
|
|
| 156 |
# Clone your fork
|
| 157 |
git clone https://github.com/YOUR_USERNAME/satellite-classification-dashboard.git
|
| 158 |
|
|
@@ -165,33 +206,40 @@ pip install -r requirements.txt
|
|
| 165 |
|
| 166 |
# Run in development mode
|
| 167 |
streamlit run app.py --logger.level=debug
|
| 168 |
-
|
|
|
|
|
|
|
| 169 |
This project is licensed under the MIT License - see the LICENSE file for details.
|
| 170 |
|
| 171 |
-
๐ Acknowledgments
|
| 172 |
-
Dataset: Spark 2021 dataset used for training
|
| 173 |
-
Frameworks: TensorFlow, Streamlit, Plotly
|
| 174 |
-
Models: Pre-trained architectures from TensorFlow/Keras
|
| 175 |
-
Hosting: Hugging Face Spaces for deployment
|
| 176 |
-
|
| 177 |
-
|
| 178 |
-
|
| 179 |
-
|
| 180 |
-
|
| 181 |
-
|
| 182 |
-
|
| 183 |
-
|
| 184 |
-
|
| 185 |
-
|
| 186 |
-
|
| 187 |
-
|
| 188 |
-
|
| 189 |
-
|
|
|
|
|
|
|
|
|
|
| 190 |
Track your application usage:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 191 |
|
| 192 |
-
Classifications Performed: Real-time counter
|
| 193 |
-
Popular Models: Usage analytics
|
| 194 |
-
Performance Metrics: Response time tracking
|
| 195 |
Built with โค๏ธ using Streamlit and TensorFlow
|
| 196 |
|
| 197 |
-
For more information, visit our Hugging Face Space
|
|
|
|
| 1 |
+
---
|
| 2 |
title: Satellite Classification Dashboard
|
| 3 |
emoji: ๐ฐ๏ธ
|
| 4 |
colorFrom: blue
|
|
|
|
| 12 |
- Bhavi23/MobilenetV2/multi_input_model_v1.keras
|
| 13 |
- Bhavi23/EfficientNet_B0/efficientnet_model.keras
|
| 14 |
- Bhavi23/DenseNet/densenet_model.keras
|
| 15 |
+
---
|
| 16 |
+
|
| 17 |
+
# ๐ฐ๏ธ Satellite Classification Dashboard
|
| 18 |
|
|
|
|
| 19 |
A comprehensive deep learning application for classifying satellite images using multiple state-of-the-art neural network models. This interactive dashboard allows you to compare different models, analyze their performance, and classify satellite images in real-time.
|
| 20 |
|
| 21 |
+
## ๐ Features
|
| 22 |
+
|
| 23 |
+
### ๐ Dashboard Overview
|
| 24 |
+
- Interactive web interface built with Streamlit
|
| 25 |
+
- Multi-page navigation system
|
| 26 |
+
- Real-time image classification
|
| 27 |
+
- Comprehensive model comparison
|
| 28 |
+
- Performance analytics and visualizations
|
| 29 |
+
|
| 30 |
+
### ๐ค Multiple Model Support
|
| 31 |
+
- **Custom CNN**: Tailored architecture for satellite imagery (95.2% accuracy)
|
| 32 |
+
- **MobileNetV2**: Lightweight model optimized for speed (92.8% accuracy, 18ms inference)
|
| 33 |
+
- **EfficientNetB0**: Best accuracy-efficiency balance (96.4% accuracy)
|
| 34 |
+
- **DenseNet121**: Complex pattern recognition capabilities (94.7% accuracy)
|
| 35 |
+
|
| 36 |
+
### ๐ฏ Classification Categories
|
| 37 |
The system can classify 11 different types of satellites and space objects:
|
| 38 |
|
| 39 |
+
- AcrimSat
|
| 40 |
+
- Aquarius
|
| 41 |
+
- Aura
|
| 42 |
+
- Calipso
|
| 43 |
+
- Cloudsat
|
| 44 |
+
- CubeSat
|
| 45 |
+
- Debris
|
| 46 |
+
- Jason
|
| 47 |
+
- Sentinel-6
|
| 48 |
+
- TRMM
|
| 49 |
+
- Terra
|
| 50 |
+
|
| 51 |
+
## ๐ Quick Start
|
| 52 |
+
|
| 53 |
+
### Try the Live Demo
|
| 54 |
Visit the Hugging Face Space to use the application directly in your browser.
|
| 55 |
|
| 56 |
+
### Local Installation
|
| 57 |
+
1. Clone the repository
|
| 58 |
+
```bash
|
| 59 |
git clone https://github.com/YOUR_USERNAME/satellite-classification-dashboard.git
|
| 60 |
cd satellite-classification-dashboard
|
| 61 |
+
```
|
| 62 |
+
|
| 63 |
+
2. Install dependencies
|
| 64 |
+
```bash
|
| 65 |
pip install -r requirements.txt
|
| 66 |
+
```
|
| 67 |
+
|
| 68 |
+
3. Run the application
|
| 69 |
+
```bash
|
| 70 |
streamlit run app.py
|
| 71 |
+
```
|
| 72 |
+
|
| 73 |
+
4. Open your browser and navigate to http://localhost:7860
|
| 74 |
+
|
| 75 |
+
## ๐ฆ Dependencies
|
| 76 |
+
- streamlit==1.47.0
|
| 77 |
+
- tensorflow-cpu==2.15.0
|
| 78 |
+
- numpy==1.26.4
|
| 79 |
+
- pandas==2.2.2
|
| 80 |
+
- plotly==5.22.0
|
| 81 |
+
- Pillow==10.4.0
|
| 82 |
+
- requests==2.32.3
|
| 83 |
+
- protobuf<4.0.0
|
| 84 |
+
|
| 85 |
+
## ๐ฎ How to Use
|
| 86 |
+
|
| 87 |
+
### 1. Home Page
|
| 88 |
+
- Overview of the system capabilities
|
| 89 |
+
- Class distribution visualization
|
| 90 |
+
- Quick navigation to other features
|
| 91 |
+
|
| 92 |
+
### 2. Model Comparison
|
| 93 |
+
- Side-by-side performance metrics
|
| 94 |
+
- Interactive charts comparing accuracy, speed, and model size
|
| 95 |
+
- Model selection recommendations
|
| 96 |
+
|
| 97 |
+
### 3. Image Classification
|
| 98 |
+
- Upload satellite images (PNG, JPG, JPEG)
|
| 99 |
+
- Select one or multiple models for prediction
|
| 100 |
+
- View confidence scores and detailed probabilities
|
| 101 |
+
- Get model recommendations based on performance
|
| 102 |
+
|
| 103 |
+
### 4. Performance Analytics
|
| 104 |
+
- Detailed performance breakdowns
|
| 105 |
+
- Efficiency metrics analysis
|
| 106 |
+
- Comprehensive model comparison tables
|
| 107 |
+
|
| 108 |
+
### 5. About Models
|
| 109 |
+
- Technical specifications for each model
|
| 110 |
+
- Strengths and use cases
|
| 111 |
+
- Model architecture details
|
| 112 |
+
|
| 113 |
+
## ๐ Model Performance
|
| 114 |
+
|
| 115 |
+
| Model | Accuracy | Precision | Recall | F1-Score | Inference Time | Model Size |
|
| 116 |
+
|-------|----------|-----------|---------|----------|----------------|------------|
|
| 117 |
+
| EfficientNetB0 | 96.4% | 96.1% | 96.2% | 96.1% | 35ms | 20.1MB |
|
| 118 |
+
| Custom CNN | 95.2% | 94.8% | 95.1% | 94.9% | 45ms | 25.3MB |
|
| 119 |
+
| DenseNet121 | 94.7% | 94.2% | 94.5% | 94.3% | 52ms | 32.8MB |
|
| 120 |
+
| MobileNetV2 | 92.8% | 92.1% | 92.5% | 92.3% | 18ms | 8.7MB |
|
| 121 |
+
|
| 122 |
+
## ๐ฏ Model Selection Guide
|
| 123 |
+
- ๐ **Best Overall Accuracy**: EfficientNetB0 (96.4%)
|
| 124 |
+
- โก **Fastest Inference**: MobileNetV2 (18ms)
|
| 125 |
+
- ๐พ **Most Lightweight**: MobileNetV2 (8.7MB)
|
| 126 |
+
- ๐ฏ **Best Balance**: EfficientNetB0 (High accuracy + efficiency)
|
| 127 |
+
- ๐ฑ **Mobile/Edge Deployment**: MobileNetV2
|
| 128 |
+
- ๐ฌ **Research/High Accuracy**: EfficientNetB0 or DenseNet121
|
| 129 |
+
|
| 130 |
+
## ๐๏ธ Architecture
|
| 131 |
+
|
| 132 |
+
### Model Sources
|
| 133 |
All models are hosted on Hugging Face Model Hub:
|
| 134 |
+
- Custom CNN: Bhavi23/Custom_CNN
|
| 135 |
+
- MobileNetV2: Bhavi23/MobilenetV2
|
| 136 |
+
- EfficientNetB0: Bhavi23/EfficientNet_B0
|
| 137 |
+
- DenseNet121: Bhavi23/DenseNet
|
| 138 |
+
|
| 139 |
+
### Data Pipeline
|
| 140 |
+
- **Image Upload**: Supports PNG, JPG, JPEG formats
|
| 141 |
+
- **Preprocessing**: Resize to 224x224, normalize to [0,1]
|
| 142 |
+
- **Model Inference**: Multi-model prediction with timing
|
| 143 |
+
- **Post-processing**: Confidence scoring and recommendations
|
| 144 |
+
|
| 145 |
+
## ๐ง Technical Details
|
| 146 |
+
|
| 147 |
+
### Input Requirements
|
| 148 |
+
- **Image Format**: PNG, JPG, JPEG
|
| 149 |
+
- **Input Size**: 224x224x3 (RGB)
|
| 150 |
+
- **Preprocessing**: Automatic resizing and normalization
|
| 151 |
+
|
| 152 |
+
### Output Format
|
| 153 |
+
- **Class Prediction**: One of 11 satellite categories
|
| 154 |
+
- **Confidence Score**: Percentage confidence (0-100%)
|
| 155 |
+
- **Inference Time**: Milliseconds for prediction
|
| 156 |
+
- **Probability Distribution**: Full softmax output for all classes
|
| 157 |
+
|
| 158 |
+
## ๐ Performance Optimization
|
| 159 |
+
- **Model Caching**: Models are cached using @st.cache_resource
|
| 160 |
+
- **Lazy Loading**: Models loaded only when needed
|
| 161 |
+
- **Efficient Preprocessing**: Optimized image pipeline
|
| 162 |
+
- **Memory Management**: Automatic cleanup of model objects
|
| 163 |
+
|
| 164 |
+
## ๐ข Deployment
|
| 165 |
+
|
| 166 |
+
### Hugging Face Spaces
|
| 167 |
The application is deployed on Hugging Face Spaces using:
|
| 168 |
+
- **Runtime**: Python 3.9
|
| 169 |
+
- **Framework**: Streamlit
|
| 170 |
+
- **Resources**: CPU-optimized for inference
|
| 171 |
|
| 172 |
+
### Docker Deployment
|
| 173 |
+
```dockerfile
|
|
|
|
|
|
|
| 174 |
FROM python:3.9-slim
|
| 175 |
|
| 176 |
WORKDIR /app
|
|
|
|
| 181 |
EXPOSE 7860
|
| 182 |
|
| 183 |
CMD ["streamlit", "run", "app.py", "--server.port=7860", "--server.address=0.0.0.0"]
|
| 184 |
+
```
|
| 185 |
+
|
| 186 |
+
## ๐ค Contributing
|
| 187 |
We welcome contributions! Please follow these steps:
|
| 188 |
|
| 189 |
+
1. Fork the repository
|
| 190 |
+
2. Create a feature branch (`git checkout -b feature/amazing-feature`)
|
| 191 |
+
3. Commit changes (`git commit -m 'Add amazing feature'`)
|
| 192 |
+
4. Push to branch (`git push origin feature/amazing-feature`)
|
| 193 |
+
5. Open a Pull Request
|
| 194 |
+
|
| 195 |
+
### Development Setup
|
| 196 |
+
```bash
|
| 197 |
# Clone your fork
|
| 198 |
git clone https://github.com/YOUR_USERNAME/satellite-classification-dashboard.git
|
| 199 |
|
|
|
|
| 206 |
|
| 207 |
# Run in development mode
|
| 208 |
streamlit run app.py --logger.level=debug
|
| 209 |
+
```
|
| 210 |
+
|
| 211 |
+
## ๐ License
|
| 212 |
This project is licensed under the MIT License - see the LICENSE file for details.
|
| 213 |
|
| 214 |
+
## ๐ Acknowledgments
|
| 215 |
+
- **Dataset**: Spark 2021 dataset used for training
|
| 216 |
+
- **Frameworks**: TensorFlow, Streamlit, Plotly
|
| 217 |
+
- **Models**: Pre-trained architectures from TensorFlow/Keras
|
| 218 |
+
- **Hosting**: Hugging Face Spaces for deployment
|
| 219 |
+
|
| 220 |
+
## ๐ Support
|
| 221 |
+
- **Issues**: GitHub Issues
|
| 222 |
+
- **Discussions**: GitHub Discussions
|
| 223 |
+
- **Email**: bhavithrass@gmail.com
|
| 224 |
+
|
| 225 |
+
## ๐ฎ Future Enhancements
|
| 226 |
+
- [ ] Real-time Video Classification
|
| 227 |
+
- [ ] Batch Processing for multiple images
|
| 228 |
+
- [ ] Model Fine-tuning Interface
|
| 229 |
+
- [ ] Custom Model Upload
|
| 230 |
+
- [ ] Export Predictions (CSV/JSON)
|
| 231 |
+
- [ ] API Endpoint for programmatic access
|
| 232 |
+
- [ ] Mobile App version
|
| 233 |
+
- [ ] Advanced Visualization with satellite orbit data
|
| 234 |
+
|
| 235 |
+
## ๐ Usage Statistics
|
| 236 |
Track your application usage:
|
| 237 |
+
- **Classifications Performed**: Real-time counter
|
| 238 |
+
- **Popular Models**: Usage analytics
|
| 239 |
+
- **Performance Metrics**: Response time tracking
|
| 240 |
+
|
| 241 |
+
---
|
| 242 |
|
|
|
|
|
|
|
|
|
|
| 243 |
Built with โค๏ธ using Streamlit and TensorFlow
|
| 244 |
|
| 245 |
+
For more information, visit our [Hugging Face Space](https://huggingface.co/spaces/Bhavi23/satellite-classification-dashboard)
|