-
Notifications
You must be signed in to change notification settings - Fork 0
Home Page
Mustafa Gurkan Canakci edited this page Sep 9, 2024
·
1 revision
This project is about a classification model that works on satellite images. We use CNN and NasNet-Mobile architectures to classify satellite images into different categories. We optimized the model's accuracy using fine-tuning, and achieved 95% accuracy.
This project aims to develop a deep learning model to classify satellite images. We improved the model performance by using pretrained models like NasNet-Mobile and fine-tuning.
- NasNet-Mobile: A pretrained model used in this project.
- CNN (Convolutional Neural Networks): Used to extract features and classify the images.
- Result: We achieved 95% accuracy after fine-tuning and optimization steps.
- The satellite images show different areas and categories.
- The images are labeled into various categories.
- The data was preprocessed to fit the CNN model.
- We applied data augmentation and normalization during training.
- Fine-tuning: We froze the last layers of the NasNet-Mobile model and retrained the rest.
- Transfer Learning: We used the NasNet-Mobile model, which was pretrained on a different dataset, and adapted it to satellite images.
- The model was trained on the training dataset and evaluated on the test dataset.
- Hyperparameter tuning (learning rate, batch size) was done to optimize the model.
To run this project, you need to install these Python libraries:
TensorFlow
Keras
OpenCV
Matplotlib
-
Clone the GitHub repository:
git clone https://github.com/grknc/Satellite-Image-Classification-CNN-Fine-Tuning-NasNet-Mobile
-
Install the required dependencies:
pip install -r requirements.txt
-
Start Jupyter notebook and open the project file:
jupyter notebook
-
Open and run the notebook file:
satellite_image_classification.ipynb
- Training accuracy: 95%
- Test accuracy: 94%
- Training time: 1 hours 2 minutes
Here’s how you can contribute to this project:
- Fork the repository.
- Make your changes.
- Create a Pull Request.
I welcome all contributions and feedback!