tahmid-ias / Faster-RCNN-fine-tune-PyTorch

Fine-Tune Faster-RCNN on a Custom Bealge Dataset using Pytorch

Usage

Train the model

python3 train.py

The trained model will be saved in the output/ with name faster-rcnn-beagle.pt

Model Inference

python3 predict.py --image path/to/test/image

#for example
python3 predict.py --image beagle/val/00000184.jpg

Note: utils.py, transforms.py, coco_eval.py, coco_utils.py, engine.py contains helper functions used during training process, and they are adopted from PyTorch Repo.