Our great sponsors
-
simple-faster-rcnn-pytorch
A simplified implemention of Faster R-CNN that replicate performance from origin paper
-
WorkOS
The modern identity platform for B2B SaaS. The APIs are flexible and easy-to-use, supporting authentication, user identity, and complex enterprise features like SSO and SCIM provisioning.
Hi all, as the title suggests I'm looking for a GitHub repo where I can edit a Faster RCNN implementation rather easily. I'm basically looking to test an idea where I have multiple branches with feature map and bounding boxes as inputs. I've modified the built-in torchvision implementation once before, but I think it's a little more complicated than I like, and I'd rather not release the entire torchvision package as part of my own work in the future. I have looked briefly into this repo https://github.com/chenyuntc/simple-faster-rcnn-pytorch/blob/master/trainer.py but it only supports a batch size of 1, and I'm not sure what it'd take to expand that capability. Is there anything better out there?
Related posts
- A simple library to train more than 20 Faster RCNN models using PyTorch (including ViTDet)
- A Library of Faster RCNN Models with Simple Training Pipeline for Custom Dataset
- PyTorch Faster RCNN Library - Support for transformer detection models.
- PyTorch Faster RCNN Library - Support for transformer detection models.
- PyTorch Faster RCNN Custom Dataset Training Made Easy