Faster/Mask RCNN RPN custom AnchorGenerator - PyTorch Forums
https://discuss.pytorch.org/t/faster-mask-rcnn-rpn-custom...16.02.2020 · Every time I define a new Anchor Generator, I get a CUDA OOM problem. I suspect it’s nothing to do with memory, there’s a weight mismatch somewhere. Here’s the code: mrcnn_args = {'num_classes':63} icdar_anchor_generator = AnchorGenerator( sizes=tuple([(4, 8, 16, 32, 64, 128, 256, 512) for r in range(5)]), aspect_ratios = tuple([(0.25, 0.5, 1, 1.5, 2) for rh in …
How could I change the RPN Head structure? - vision - PyTorch ...
discuss.pytorch.org › t › how-could-i-change-the-rpnMay 08, 2020 · It uses 3 anchors_per_location, but I think you’re right.It is a very similar implementation. However, while PyTorch’s implementation returns a logit tensor in the format [batch, anchors, height, width], the one I am using returns a logit tensor in the format [batch, anchors per location * 2, height, width].