OSS build mask head using fbnet builder

Summary:
[WIP] Will add pretrained weights and update model url & scores

build mask head using fbnet builder and retrain weights

Reviewed By: wat3rBro

Differential Revision: D27992340

fbshipit-source-id: a216a99954eb3784438d595cd09cbb19e70ec3c3

MODEL_ZOO.md

@@ -47,8 +47,8 @@ FBNet series are efficient mobile backbones discovered via neural architecture s
 
 | name                                                         | box AP | mask AP | model id  | download                                                     |
 | ------------------------------------------------------------ | ------ | ------- | --------- | ------------------------------------------------------------ |
-| [Mask-RCNN-FBNetV3A](./configs/mask_rcnn_fbnetv3a_C4.yaml)   | 23.74  | 21.18   | 250355374 | [model](https://mobile-cv.s3-us-west-2.amazonaws.com/d2go/models/250355374/model_0479999.pth) \|[metrics](https://mobile-cv.s3-us-west-2.amazonaws.com/d2go/models/250355374/metrics.json) |
-| [Mask-RCNN-FBNetV3A-dsmask](./configs/mask_rcnn_fbnetv3a_dsmask_C4.yaml) | 21.81  | 19.76   | 250414867 | [model](https://mobile-cv.s3-us-west-2.amazonaws.com/d2go/models/250414867/model_0399999.pth) \|[metrics](https://mobile-cv.s3-us-west-2.amazonaws.com/d2go/models/250414867/metrics.json) |
+| [Mask-RCNN-FBNetV3A](./configs/mask_rcnn_fbnetv3a_C4.yaml)   | 23.05  | 20.71   | 268421013 | [model](https://mobile-cv.s3-us-west-2.amazonaws.com/d2go/models/268421013/model_final.pth) \|[metrics](https://mobile-cv.s3-us-west-2.amazonaws.com/d2go/models/268421013/metrics.json) |
+| [Mask-RCNN-FBNetV3A-dsmask](./configs/mask_rcnn_fbnetv3a_dsmask_C4.yaml) | 21.76  | 19.97   | 268412271 | [model](https://mobile-cv.s3-us-west-2.amazonaws.com/d2go/models/268412271/model_0499999.pth) \|[metrics](https://mobile-cv.s3-us-west-2.amazonaws.com/d2go/models/268412271/metrics.json) |
 | [Mask-RCNN-FBNetV3G-FPN](./configs/mask_rcnn_fbnetv3g_fpn.yaml) | 43.88  | 39.25   | 250376154 | [model](https://mobile-cv.s3-us-west-2.amazonaws.com/d2go/models/250376154/model_0404999.pth) \|[metrics](https://mobile-cv.s3-us-west-2.amazonaws.com/d2go/models/250376154/metrics.json) |
 
 ### COCO Person Keypoint Detection

configs/mask_rcnn_fbnetv3a_C4.yaml

@@ -28,8 +28,7 @@ MODEL:
     POOLER_RESOLUTION: 6
     NORM: "naiveSyncBN"
   ROI_MASK_HEAD:
-    NAME: "MaskRCNNConvUpsampleHead"
-    NUM_CONV: 4
+    NAME: "FBNetV2RoIMaskHead"
     POOLER_RESOLUTION: 14
     NORM: "naiveSyncBN"
 MODEL_EMA:

d2go/modeling/modeldef/modeldef.py

@@ -443,6 +443,7 @@ MODEL_ARCH_BUILTIN = {
         "trunk": FBNetV3_A[0:4],
         "rpn": [[_repeat_last(FBNetV3_A[3])]],
         "bbox": [FBNetV3_A[4]],
+        "mask": SMALL_DS_UPSAMPLE_HEAD_STAGES,
         "basic_args": _BASIC_ARGS,
     },
     "FBNetV3_B": {