From 159697a26739d71636ed1d5c31de3471704fca53 Mon Sep 17 00:00:00 2001 From: Xianzhi Du Date: Sun, 23 Jan 2022 21:57:57 -0800 Subject: [PATCH] Internal change PiperOrigin-RevId: 423725954 --- official/vision/beta/MODEL_GARDEN.md | 1 + ...asrcnn_tpu.yaml => coco_spinenet96_cascadercnn_tpu.yaml} | 6 ++++-- 2 files changed, 5 insertions(+), 2 deletions(-) rename official/vision/beta/configs/experiments/maskrcnn/{coco_spinenet96_casrcnn_tpu.yaml => coco_spinenet96_cascadercnn_tpu.yaml} (88%) diff --git a/official/vision/beta/MODEL_GARDEN.md b/official/vision/beta/MODEL_GARDEN.md index 1546fbedb..ebb0cf280 100644 --- a/official/vision/beta/MODEL_GARDEN.md +++ b/official/vision/beta/MODEL_GARDEN.md @@ -137,6 +137,7 @@ evaluated on [COCO](https://cocodataset.org/) val2017. | Backbone | Resolution | Epochs | Params (M) | Box AP | Mask AP | Download ------------ | :--------: | -----: | ---------: | -----: | ------: | -------: | 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)| +| 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)| | 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)| ## Semantic Segmentation diff --git a/official/vision/beta/configs/experiments/maskrcnn/coco_spinenet96_casrcnn_tpu.yaml b/official/vision/beta/configs/experiments/maskrcnn/coco_spinenet96_cascadercnn_tpu.yaml similarity index 88% rename from official/vision/beta/configs/experiments/maskrcnn/coco_spinenet96_casrcnn_tpu.yaml rename to official/vision/beta/configs/experiments/maskrcnn/coco_spinenet96_cascadercnn_tpu.yaml index 612608333..b8dd2fb9d 100644 --- a/official/vision/beta/configs/experiments/maskrcnn/coco_spinenet96_casrcnn_tpu.yaml +++ b/official/vision/beta/configs/experiments/maskrcnn/coco_spinenet96_cascadercnn_tpu.yaml @@ -1,3 +1,5 @@ +# --experiment_type=cascadercnn_spinenet_coco +# Expect to reach: box mAP: 51.9%, mask mAP: 45.0% on COCO runtime: distribution_strategy: 'tpu' mixed_precision_dtype: 'bfloat16' @@ -8,12 +10,12 @@ task: parser: aug_rand_hflip: true aug_scale_min: 0.1 - aug_scale_max: 2.0 + aug_scale_max: 2.5 losses: l2_weight_decay: 0.00004 model: anchor: - anchor_size: 3.0 + anchor_size: 4.0 num_scales: 3 min_level: 3 max_level: 7 -- GitLab