This project classifies images of different weather conditions into 11 categories:
dew, fogsmog, frost, glaze, hail, lightning, rain, rainbow, rime, sandstorm, snow.
- Custom dataset stored in
./data/weather class/dataset/ - Each folder corresponds to one weather category
- No predefined train/test split β split done using
validation_splitin TensorFlow
-
Custom CNN (Convolutional Neural Network)
- 3 Conv2D layers + Dense layers
- Accuracy: ~17% (not great)
-
MobileNetV2 (Transfer Learning)
- Pretrained on ImageNet
- Fine-tuned for weather dataset
- Accuracy: ~80β85% after tuning
| Model | Accuracy |
|---|---|
| Custom CNN | ~17% |
| MobileNetV2 | ~80β85% |
- Clone the repository
git clone https://github.com/Vision-Ali/Weather-Classification-CNN.git cd Weather-Classification-CNN