Skip to content

Commit 159697a

Browse files
xianzhidutensorflower-gardener
authored andcommitted
Internal change
PiperOrigin-RevId: 423725954
1 parent 78a4123 commit 159697a

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

official/vision/beta/MODEL_GARDEN.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -137,6 +137,7 @@ evaluated on [COCO](https://cocodataset.org/) val2017.
137137
| Backbone | Resolution | Epochs | Params (M) | Box AP | Mask AP | Download
138138
------------ | :--------: | -----: | ---------: | -----: | ------: | -------:
139139
| SpineNet-49 | 640x640 | 500 | 56.4 | 46.4 | 40.0 | [config](https://github.com/tensorflow/models/blob/master/official/vision/beta/configs/experiments/maskrcnn/coco_spinenet49_cascadercnn_tpu.yaml)|
140+
| SpineNet-96 | 1024x1024 | 500 | 70.8 | 50.9 | 43.8 | [config](https://github.com/tensorflow/models/blob/master/official/vision/beta/configs/experiments/maskrcnn/coco_spinenet96_cascadercnn_tpu.yaml)|
140141
| SpineNet-143 | 1280x1280 | 500 | 94.9 | 51.9 | 45.0 | [config](https://github.com/tensorflow/models/blob/master/official/vision/beta/configs/experiments/maskrcnn/coco_spinenet143_cascadercnn_tpu.yaml)|
141142

142143
## Semantic Segmentation

official/vision/beta/configs/experiments/maskrcnn/coco_spinenet96_casrcnn_tpu.yaml renamed to official/vision/beta/configs/experiments/maskrcnn/coco_spinenet96_cascadercnn_tpu.yaml

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
1+
# --experiment_type=cascadercnn_spinenet_coco
2+
# Expect to reach: box mAP: 51.9%, mask mAP: 45.0% on COCO
13
runtime:
24
distribution_strategy: 'tpu'
35
mixed_precision_dtype: 'bfloat16'
@@ -8,12 +10,12 @@ task:
810
parser:
911
aug_rand_hflip: true
1012
aug_scale_min: 0.1
11-
aug_scale_max: 2.0
13+
aug_scale_max: 2.5
1214
losses:
1315
l2_weight_decay: 0.00004
1416
model:
1517
anchor:
16-
anchor_size: 3.0
18+
anchor_size: 4.0
1719
num_scales: 3
1820
min_level: 3
1921
max_level: 7

0 commit comments

Comments
 (0)