Description:

pytorchfasterrcnn 0.2.1

Pytorch Faster RCNN






Faster RCNN Fine-Tune Implementation in Pytorch.
How to use ?

git clone the repo

git clone https://github.com/oke-aditya/pytorch_fasterrcnn.git


Install PyTorch and torchvision for your system.

Simply edit the config file to set your hyper parameters.

Keep the training and validation csv file as follows

NOTE
Do not use target as 0 class. It is reserved as background.
image_id xtl ytl xbr ybr target
1 xmin ymin xmax ymax 1
1 xmin ymin xmax ymax 2
2 xmin ymin xmax ymax 3



Simply edit the config file to set your hyper parameters


Run the train.py file


Features: -

It works for multiple class object detection.

Backbones Supported: -


Note that backbones are pretrained on imagenet.


Following backbones are supported



vgg11, vgg13, vgg16, vgg19
resnet18, resnet34, resnet50, resnet101, resnet152
renext101
mobilenet_v2

Sample Outputs
Helmet Detector

Mask Detector

If you like the implemenation or have taken an inspiration do give a star :-)

License

For personal and professional use. You cannot resell or redistribute these repositories in their original state.

Share

• Share on Facebook
• Share on Twitter