/Pytorch-Classification-Net-Template

a template for classification task based on pytorch

Primary LanguagePython

Project Description

This project is designed for quickly generating a pytorch project for simple image classification task.

After download the whole project, all you need to modify is the parameters in train.py, including data path, network type, training parameters, model path and so on.

File Description

dataset.py

implement dataset and dataloader

network.py

implement classification network

train.py

implement a pipeline to train the network

test.py

implement test process based on trained model