Faster RCNN Fine-Tuner in Pytorch
This repository provides codebase that can be used to fine-tune existing FRCNN in torchvision.
It provides a powerful engine that can be used to train FRCNN using simple train_step functions.
Backbones supported
Following backbones are supported
- vgg11, vgg13, vgg16, vgg19
- resnet18, resnet34, resnet50, resnet101, resnet152
- renext101
- mobilenet_v2
More stuff coming in next release 👍