How do I add a new model?

❓ Questions and Help

I want to add a new model which is basically Res101 maskrccn without the FPN. Does setting USE_FPN=False here https://github.com/facebookresearch/maskrcnn-benchmark/blob/master/configs/e2e_mask_rcnn_R_101_FPN_1x.yaml ensure no use of FPN?

Would I need to change any other files? If I were to make a new .yaml file with the name e2e_mask_rcnn_R_101_1x.yaml what changes would I need to do?

Currently, I have made a new .yaml file and set USE_FPN=False, and Anchor_Strides=(16,) as per default.

However, this gives me channel mismatch error RuntimeError: Given transposed=1, weight of size [2048, 256, 2, 2], expected input[4, 256, 14, 14] to have 2048 channels, but got 256 channels instead on line: maskrcnn-benchmark/maskrcnn_benchmark/modeling/roi_heads/mask_head/roi_mask_predictors.py", line 35, in forward x = F.relu(self.conv5_mask(x))

[wesleylp]

Hi @fmassa, thanks for your quick answer! I’ve seen #346 handled the R-101-C4 backbone registration but I was unsure about ROI_BOX_HEAD.FEATURE_EXTRACTOR parameter. Thank you, again!

[fmassa]

Because FPN models are generally faster, because their HEAD is much smaller than the default HEAD from C4 models.