simple-faster-rcnn-pytorch VS Yet-Another-EfficientDet-Pytorch

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?

However, there are papers on oriented object detectors (see https://arxiv.org/pdf/1911.07732.pdf) for example. In that paper, they do achieve better results using oriented bounding boxes. If you want to go down that route, I would suggest using the EfficientDet model, because the PyTorch code that you'll find for it is quite easy to understand and modify. For example, I've taken https://github.com/zylo117/Yet-Another-EfficientDet-Pytorch, and modified it to include a "thing-ness" logit, and this was pretty easy to do. Classic EfficientDet models only include logits (aka output neurons that get softmax-ed) for each class, and if any one of these class neurons is greater than 0.5, then it is considered "a thing". Anyway - that's digression, but my point is that I've thought about adding oriented box support to an EfficientDet model, and it didn't seem to be too hard, although I haven't actually done it. If I was to start now, I would probably go with https://github.com/rwightman/efficientdet-pytorch, since Ross Wightman's models are becoming a de-facto standard in the PyTorch world for all things image-related.