README.md

Image Classification Demo

This project mainly modified pytorch official image classification tutorial into a python package format. It can serve as a start-up repo for any image classification task.

Enviroments

• Docker: https://hub.docker.com/r/pytorch/pytorch

Sources

• Data: https://download.pytorch.org/tutorial/hymenoptera_data.zip
• Tutorial: https://pytorch.org/tutorials/beginner/transfer_learning_tutorial.html