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.
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