How to add two MaskRCNN heads on Same backbone?

[123mw123]

I want to train a model with same backbone( R50-FPN) but two MaskRCNN heads. Can somebody help me or point me to some resources on how to do this change?

[ppwwyyxx]

StandardROIHeads

detectron2/detectron2/modeling/roi_heads/roi_heads.py

Line 521 in 3c0509e

class StandardROIHeads(ROIHeads):

is where all heads are added. It can be extended to add other heads, e.g. https://github.com/facebookresearch/meshrcnn/tree/master/meshrcnn/modeling/roi_heads adds a lot of heads