Welcome to Weather Image Recognition! This project utilizes machine learning and advanced neural network architectures to classify weather conditions from images. 🌤️🌧️❄️
- 📂 Data Handling: Load, preprocess, and augment weather image datasets.
- 🖼️ Image Classification: Classify weather types (e.g., sunny, rainy, snowy) with high accuracy.
- 🧠 Architectures Used:
- Custom CNN: A lightweight convolutional neural network designed specifically for this task.
- Torchvision Models: Integrates pre-trained models for transfer learning (e.g., ResNet, VGG).
- 📈 Evaluation Metrics: Analyze model performance with accuracy, confusion matrices, and visualizations.
- 🔧 Customizable Pipelines: Easily adapt the notebook for different datasets or tasks.
- 🐍 Python 3.7+
- 📦 Required libraries (install with
pip
):torch
torchvision
numpy
pandas
matplotlib
- Clone the repository:
git clone https://github.com/your-username/weather-image-recognition.git
- Navigate to the project folder:
cd weather-image-recognition
- Install dependencies:
pip install -r requirements.txt
- Open the notebook:
jupyter notebook Weather_Image_Recognition.ipynb
- Run the cells: Follow the step-by-step guidance in the notebook.
- Load and preprocess labeled weather images. 🌅🌩️🌨️
- Train a custom CNN or fine-tune a pre-trained model (like ResNet).
- Evaluate model performance using test datasets.
- Visualize predictions and analyze errors.
- Designed for lightweight yet effective classification.
- Includes convolutional layers, max-pooling, and fully connected layers optimized for weather data.
- Uses
torchvision.models
for transfer learning. - Examples: ResNet, VGG, or other deep CNNs, fine-tuned on weather datasets.