This repository contains training and inference scripts for the EVRAS track 2 (computer vision) solution by Ammar Ali.
Don't forget to download the data
-
pip install -r requirements.txt
- Download weights from here
-
python run.py --weights weights_path --data images_path --output output-dir --sub_sample sample_submission.json --imgsz image_size
-
docker build .
-
pip install -r requirements.txt
- Download weights from here
-
python run.py --img_path orig1.jpg --weights tf_efficientnet_b1_ns_fold0_best.pth
-
docker build .
