Image Classification (PyTorch)

---

---

---

Train / Test

1. pip install -r requirements.txt
2. python train.py

Dataset format

The dataset is not public.

dataroot
├── train
    ├── 1 (quokkas)
        ├── 0001.jpg
        ├── 0002.jpg
        ├── ...
    ├── 2 (squirrels)
        ├── 0001.jpg
        ├── 0002.jpg
        ├── ...
├── test
    ├── 1 (quokkas)
        ├── 0001.jpg
        ├── 0002.jpg
        ├── ...
    ├── 2 (squirrels)
        ├── 0001.jpg
        ├── 0002.jpg
        ├── ...

Result of training with the dataset