Commit 30e798a6 authored by Hang Zhang's avatar Hang Zhang Committed by Facebook GitHub Bot
Browse files

fix model_zoo links & retrain V3G mask rcnn

Summary:
Pull Request resolved: https://github.com/facebookresearch/d2go/pull/102

- fix model_zoo model urls (missed in D27992340 (https://github.com/facebookresearch/d2go/commit/477ab964e2165cb586b5c00425f6e463d7edeadd))
- update mask rcnn fbnet V3G config
- update v3g retrained weights

Reviewed By: ppwwyyxx, wat3rBro

Differential Revision: D29627615

fbshipit-source-id: 0694772e47b9c58965e47492177a5d6de53364cb
parent a4f06b88
...@@ -49,7 +49,7 @@ FBNet series are efficient mobile backbones discovered via neural architecture s ...@@ -49,7 +49,7 @@ FBNet series are efficient mobile backbones discovered via neural architecture s
| ------------------------------------------------------------ | ------ | ------- | --------- | ------------------------------------------------------------ | | ------------------------------------------------------------ | ------ | ------- | --------- | ------------------------------------------------------------ |
| [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](./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-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.31 | 39.24 | 287445123 | [model](https://mobile-cv.s3-us-west-2.amazonaws.com/d2go/models/287445123/model_0409999.pth) \|[metrics](https://mobile-cv.s3-us-west-2.amazonaws.com/d2go/models/287445123/metrics.json) |
### COCO Person Keypoint Detection ### COCO Person Keypoint Detection
......
...@@ -35,8 +35,7 @@ MODEL: ...@@ -35,8 +35,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:
......
...@@ -13,12 +13,12 @@ class _ModelZooUrls(object): ...@@ -13,12 +13,12 @@ class _ModelZooUrls(object):
""" """
S3_PREFIX = "https://mobile-cv.s3-us-west-2.amazonaws.com/d2go/models/" S3_PREFIX = "https://mobile-cv.s3-us-west-2.amazonaws.com/d2go/models/"
CONFIG_PATH_TO_URL_SUFFIX = { CONFIG_PATH_TO_URL_SUFFIX = {
"faster_rcnn_fbnetv3a_C4.yaml": "246823121/model_0479999.pth", "faster_rcnn_fbnetv3a_C4.yaml": "268421013/model_final.pth",
"faster_rcnn_fbnetv3a_dsmask_C4.yaml": "250414811/model_0399999.pth", "faster_rcnn_fbnetv3a_dsmask_C4.yaml": "268412271/model_0499999.pth",
"faster_rcnn_fbnetv3g_fpn.yaml": "250356938/model_0374999.pth", "faster_rcnn_fbnetv3g_fpn.yaml": "250356938/model_0374999.pth",
"mask_rcnn_fbnetv3a_C4.yaml": "250355374/model_0479999.pth", "mask_rcnn_fbnetv3a_C4.yaml": "250355374/model_0479999.pth",
"mask_rcnn_fbnetv3a_dsmask_C4.yaml": "250414867/model_0399999.pth", "mask_rcnn_fbnetv3a_dsmask_C4.yaml": "250414867/model_0399999.pth",
"mask_rcnn_fbnetv3g_fpn.yaml": "250376154/model_0404999.pth", "mask_rcnn_fbnetv3g_fpn.yaml": "287445123/model_0409999.pth",
"keypoint_rcnn_fbnetv3a_dsmask_C4.yaml": "250430934/model_0389999.pth", "keypoint_rcnn_fbnetv3a_dsmask_C4.yaml": "250430934/model_0389999.pth",
} }
......
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