Classify between a Dog and Cat in each image
1. Võ Đình Đạt | Dat.vd2148902sis.hust.edu.vn
2. Trần Quốc Đệ | [email protected]
3. Phạm Quang Nguyên Hoàng | [email protected]
4. Đường Minh Quân | [email protected]
5. Phạm Ngọc Quân | [email protected]
| .gitignore
│ kaggle.json
│ LICENSE
│ README.md
│ requirements.txt
│
├───demo
│ demo_main.py
│ demo_model.py
│ model.pt
│ resnet.pt
│
├───img
│ test.png
│
└───src
Augmentation_Experiment.ipynb
Cat_Dog.ipynb
EDA.ipynb
https://www.kaggle.com/competitions/dog-vs-cat-classification
-
Step 2: Save a copy in order to run.
-
Step 3: Run notebook (prefer GPU runtime and High RAM for faster training).
- Install requirement:
git clone https://github.com/datvodinh10/cat-dog-classification.git
pip install requirements.txt
-
Open
src/Cat_Dog.ipynb
-
Press
Run all
NOTE: Require GPU or at least 32Gb RAM and CPU to run local.