Commit 477ab964 authored by Hang Zhang's avatar Hang Zhang Committed by Facebook GitHub Bot
Browse files

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
parent e84d3414
...@@ -47,8 +47,8 @@ FBNet series are efficient mobile backbones discovered via neural architecture s ...@@ -47,8 +47,8 @@ FBNet series are efficient mobile backbones discovered via neural architecture s
| name | box AP | mask AP | model id | download | | 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](./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.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-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) | | [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 ### COCO Person Keypoint Detection
......
...@@ -28,8 +28,7 @@ MODEL: ...@@ -28,8 +28,7 @@ MODEL:
POOLER_RESOLUTION: 6 POOLER_RESOLUTION: 6
NORM: "naiveSyncBN" NORM: "naiveSyncBN"
ROI_MASK_HEAD: ROI_MASK_HEAD:
NAME: "MaskRCNNConvUpsampleHead" NAME: "FBNetV2RoIMaskHead"
NUM_CONV: 4
POOLER_RESOLUTION: 14 POOLER_RESOLUTION: 14
NORM: "naiveSyncBN" NORM: "naiveSyncBN"
MODEL_EMA: MODEL_EMA:
......
...@@ -443,6 +443,7 @@ MODEL_ARCH_BUILTIN = { ...@@ -443,6 +443,7 @@ MODEL_ARCH_BUILTIN = {
"trunk": FBNetV3_A[0:4], "trunk": FBNetV3_A[0:4],
"rpn": [[_repeat_last(FBNetV3_A[3])]], "rpn": [[_repeat_last(FBNetV3_A[3])]],
"bbox": [FBNetV3_A[4]], "bbox": [FBNetV3_A[4]],
"mask": SMALL_DS_UPSAMPLE_HEAD_STAGES,
"basic_args": _BASIC_ARGS, "basic_args": _BASIC_ARGS,
}, },
"FBNetV3_B": { "FBNetV3_B": {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment