/pytorch_classification

Multi-class weather classification in PyTorch

Primary LanguagePython

Multi-class weather classification in PyTorch

Link to the original dataset

Requirements

Before running the scripts, all the necessary packages need to be installed

pip install -r requirements.txt

How to get started

First of all, it is necessary to run the script to clear the data and delete pictures of an unsuitable format

Then run the main file

Graphs from TensorBoard