Skip to content

Image classification on Satellite Dataset-RSI-CB256 with torchvision models.

Notifications You must be signed in to change notification settings

dilaraozdemir/satellite-image-classification-pytorch

Repository files navigation

Image Classification with Python on Satellite Image Classification Dataset-RSI-CB256

NOTE 📝 This dataset is taken from kaggle.

Dataset Explanation

❗ In this dataset;

There are 4 classes: "cloud", "desert", "green_area" and "water".

While obtaining the data, the Google map snapshots were mixed with the data received from the sensors.

Visualization samples of dataset

🔸 Cloud

train_352train_352train_352

🔸 Desert

train_352train_352train_352

🔸 Green Area

train_352train_352train_352

🔸 Water

train_352train_352train_352

Usage

Creating Anaconda Environment

conda create -n torchgpu python=3.6 
conda activate torchgpu 

Pytorch Installation

You can install pytorch by running the following commands:

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

❗ In order for the command to work, NVIDIA's distributions for cuda packages must be installed on your local env.

Releases

No releases published

Packages

No packages published