This program is the implementation of faster RCNN is pytorch with simple steps:
- cretae a folder like sample (Traffic_sign) containing images (test and train) ,labels (test, train), model and config.yaml Note: if the labels are in yolo format you can use available script here (yolo to csv) and change the format to pascal.
- edit the config file with correct paths
- run the faster_RCNN_train_general.py to train the model.
- run the faster_RCNN_detect_general.py to use the model for detections.