The final project for COMP755.
- python >= 3.6.9
- Install dependencies
pip install -r requirements.txt
- Download our synthetic dataset here
- Finetune pretrained inception-v3 using our data.
Finetuning the pretrained model
python train_finetune.py --gpu 0
Inference using finetuned model
python inference_finetune.py
- Inference using model pretrained on ImageNet.
python inference.py --gpu 0