Using this repo you can easily train Faster R-CNN model on google colab.
- Clone the repo in google colab using
!git clone {COPY THE HTTP LINK FOR THIS REPO}
- Upload your data to the data directory(Directory already has a format in place, you can utilize that).
- Make changes in the config.py file(Update the Dataset paths in the config file to yours, also change hyperparameters such as epochs etc to your desired needs).
- Run
!python train.py
to train the model - Run
!python inference.py
to test the model on test data(Data in test directory)
Repo was created from reference of this Web : https://debuggercafe.com/a-simple-pipeline-to-train-pytorch-faster-rcnn-object-detection-model/