from here!
┌── requirements.txt
├── test/
│ ├── ...
| └── .jpg
├── train/
| ├── cat/
│ │ ├── ...
| │ └── .jpg
│ └── dog/
│ ├── ...
| └── .jpg
├── models/
├── dataset.py
├── infer.py
├── main.py
├── random_erasing.py
└── train.py
python main.py
python train.py
- We use ResNet50 as pretrained model.