Classification with Pytorch Networks

Installation:

Install Python 3.10.10

Install CUDA Toolkit 11.8

Install Pytorch 2.0

Install Other Requeriments: pip install -r requeriments.txt

Organized Your Dataset:

|-- Dataset_Name/
|   |-- train/
|   |   |-- label_1/
|   |   |   |-- image1.jpg
|   |   |   |-- image2.jpg
|   |   |   |-- image3.jpg
|   |   |   |-- ...
|   |   |-- label_2/
|   |   |   |-- image1.jpg
|   |   |   |-- image2.jpg
|   |   |   |-- image3.jpg
|   |   |   |-- ...
|   |   |-- label_3/
|   |   |   |-- image1.jpg
|   |   |   |-- image2.jpg
|   |   |   |-- image3.jpg
|   |   |   |-- ...
|   |   |-- ...
|   |-- test/
|   |   |-- label_1/
|   |   |   |-- image1.jpg
|   |   |   |-- image2.jpg
|   |   |   |-- image3.jpg
|   |   |   |-- ...
|   |   |-- label_2/
|   |   |   |-- image1.jpg
|   |   |   |-- image2.jpg
|   |   |   |-- image3.jpg
|   |   |   |-- ...
|   |   |-- label_3/
|   |   |   |-- image1.jpg
|   |   |   |-- image2.jpg
|   |   |   |-- image3.jpg
|   |   |   |-- ...
|   |   |-- ...
|   |-- valid/
|   |   |-- label_1/
|   |   |   |-- image1.jpg
|   |   |   |-- image2.jpg
|   |   |   |-- image3.jpg
|   |   |   |-- ...
|   |   |-- label_2/
|   |   |   |-- image1.jpg
|   |   |   |-- image2.jpg
|   |   |   |-- image3.jpg
|   |   |   |-- ...
|   |   |-- label_3/
|   |   |   |-- image1.jpg
|   |   |   |-- image2.jpg
|   |   |   |-- image3.jpg
|   |   |   |-- ...
|   |   |-- ...
...

Execute:

Train:

python trainer.py --dataset dataset_path --output output_folder --size images_size --epochs num_epochs --batch_size batch_size --model model_name --learning_rate learning_rate

Test: