Image classification in PyTorch
Prerequisites
- PyTorch 0.4+
- Python 3.5+
Datasets
Put your datasets in ./data/train/. As follows:
├─data
│ ├─abyssinian
│ │ Abyssinian_0.jpg
│ │ Abyssinian_1.jpg
│ │ ...
│ │
│ ├─american_bulldog
│ │ american_bulldog_0.jpg
│ │ american_bulldog_1.jpg
│ │ ...
│ │
│ ├─Birman
│ │ Birman_0.jpg
│ │ Birman_1.jpg
│ │ ...
│ │
│ ├─Sphynx
│ │ Sphynx_0.jpg
│ │ Sphynx_1.jpg
│ │ ...
│ │
│ └─yorkshire_terrier
│ yorkshire_terrier_0.jpg
│ yorkshire_terrier_1.jpg
│ ...
Usage
Train your classifier: $ python train.py.
Test your classifier: $ python test.py.