For running the model we have removed some of the images base on the following criteria:
- If bounding box dimensions are negative.
- If there are no bounding boxes available.
- coordinates of bounding boxes are same.
Make sure all libraries are installed.
To run different Ablation studies:
- Resent 50 : sh run_resnet.sh
- VGG 10 : sh run_vgg.sh
- MobileNet v2 : sh run_mobilenet.sh
After running the models the best model will be saved in results folder then run the following script file to get the results on test images:
- sh run_inference.sh resnet
- sh run_inference.sh vgg
- sh run_inference.sh mobile_net
The YOLOv5 training is inspired from this repo
To run the YOLO models for training and inference, see yolov5/yolov5.ipynb