From 554166437fc64865a38bf0bf2acd5c418476820a Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Tue, 13 Oct 2020 13:21:56 -0700 Subject: [PATCH] Internal change PiperOrigin-RevId: 336942308 --- official/vision/beta/MODEL_GARDEN.md | 11 +++++++---- .../experiments/retinanet/coco_spinenet143_tpu.yaml | 3 ++- .../experiments/retinanet/coco_spinenet49_tpu.yaml | 1 + .../experiments/retinanet/coco_spinenet96_tpu.yaml | 1 + 4 files changed, 11 insertions(+), 5 deletions(-) diff --git a/official/vision/beta/MODEL_GARDEN.md b/official/vision/beta/MODEL_GARDEN.md index 3b9d5fe20..d47aa2ee9 100644 --- a/official/vision/beta/MODEL_GARDEN.md +++ b/official/vision/beta/MODEL_GARDEN.md @@ -49,12 +49,15 @@ TF Vision model garden provides a large collection of baselines and checkpoints | R50-FPN | 640x640 | 12 | 97.0 | 34.0 | 34.3 | config| | R50-FPN | 640x640 | 36 | 97.0 | 34.0 | 37.3 | config| -#### RetinaNet (Trained from scratch) +#### RetinaNet (Trained from scratch) with training features including: +* Stochastic depth with drop rate 0.2. +* Swish activation. + | backbone | resolution | epochs | FLOPs (B) | params (M) | box AP | download | | ------------ |:-------------:| ---------:|-----------:|--------:|---------:|-----------:| -| SpineNet-49 | 640x640 | 350 | 85.4| 28.5 | 42.4| config| -| SpineNet-96 | 1024x1024 | 350 | 265.4 | 43.0 | 46.0 | config | -| SpineNet-143 | 1280x1280 | 350 | 524.0 | 67.0 | 46.8 |config| +| SpineNet-49 | 640x640 | 500 | 85.4| 28.5 | 44.2 | [config](https://github.com/tensorflow/models/blob/master/official/vision/beta/configs/experiments/retinanet/coco_spinenet49_tpu.yaml)| +| SpineNet-96 | 1024x1024 | 500 | 265.4 | 43.0 | 48.5 | [config](https://github.com/tensorflow/models/blob/master/official/vision/beta/configs/experiments/retinanet/coco_spinenet96_tpu.yaml) | +| SpineNet-143 | 1280x1280 | 500 | 524.0 | 67.0 | 50.0 | [config](https://github.com/tensorflow/models/blob/master/official/vision/beta/configs/experiments/retinanet/coco_spinenet143_tpu.yaml)| ### Instance Segmentation Baselines diff --git a/official/vision/beta/configs/experiments/retinanet/coco_spinenet143_tpu.yaml b/official/vision/beta/configs/experiments/retinanet/coco_spinenet143_tpu.yaml index 7c6a37db8..919163f2c 100644 --- a/official/vision/beta/configs/experiments/retinanet/coco_spinenet143_tpu.yaml +++ b/official/vision/beta/configs/experiments/retinanet/coco_spinenet143_tpu.yaml @@ -1,9 +1,10 @@ +# SpineNet-143 COCO detection with protocal C config. Expecting 50.0% AP. runtime: distribution_strategy: 'tpu' mixed_precision_dtype: 'bfloat16' task: annotation_file: '/readahead/200M/placer/prod/home/snaggletooth/test/data/coco/instances_val2017.json' - gradient_clip_norm: 0.0 + gradient_clip_norm: 10.0 losses: l2_weight_decay: 4.0e-05 model: diff --git a/official/vision/beta/configs/experiments/retinanet/coco_spinenet49_tpu.yaml b/official/vision/beta/configs/experiments/retinanet/coco_spinenet49_tpu.yaml index dbb866209..b97867b6c 100644 --- a/official/vision/beta/configs/experiments/retinanet/coco_spinenet49_tpu.yaml +++ b/official/vision/beta/configs/experiments/retinanet/coco_spinenet49_tpu.yaml @@ -1,3 +1,4 @@ +# SpineNet-49 COCO detection with protocal C config. Expecting 44.2% AP. runtime: distribution_strategy: 'tpu' mixed_precision_dtype: 'bfloat16' diff --git a/official/vision/beta/configs/experiments/retinanet/coco_spinenet96_tpu.yaml b/official/vision/beta/configs/experiments/retinanet/coco_spinenet96_tpu.yaml index 58049f333..68cb946ab 100644 --- a/official/vision/beta/configs/experiments/retinanet/coco_spinenet96_tpu.yaml +++ b/official/vision/beta/configs/experiments/retinanet/coco_spinenet96_tpu.yaml @@ -1,3 +1,4 @@ +# SpineNet-96 COCO detection with protocol C config. Expecting 48.5% AP. runtime: distribution_strategy: 'tpu' mixed_precision_dtype: 'bfloat16' -- GitLab