Skip to content

GitLab

"docs/zh_cn/vscode:/vscode.git/clone" did not exist on "1f161f685979ab2ca929152e6a577b5ac8ddf2eb"
Commit 1cb5e572 authored by Xianzhi Du's avatar Xianzhi Du Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 367083514
parent 9fe85228
Show whitespace changes
Inline Side-by-side
Showing with 15 additions and 6 deletions
official/vision/beta/MODEL_GARDEN.md
View file @ 1cb5e572
...@@ -67,9 +67,15 @@ ResNet-RS-350 | 320x320 | 164.3 | 84.2 | 96.9 | [config](https://github.c ...@@ -67,9 +67,15 @@ ResNet-RS-350 | 320x320 | 164.3 | 84.2 | 96.9 | [config](https://github.c
| backbone | resolution | epochs | FLOPs (B) | params (M) | box AP | download | | backbone | resolution | epochs | FLOPs (B) | params (M) | box AP | download |
| ------------ |:-------------:| ---------:|-----------:|--------:|---------:|-----------:| | ------------ |:-------------:| ---------:|-----------:|--------:|---------:|-----------:|
| 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) [TB.dev](https://tensorboard.dev/experiment/n2UN83TkTdyKZn3slCWulg/#scalars&_smoothingWeight=0)| | 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) \| [TB.dev](https://tensorboard.dev/experiment/n2UN83TkTdyKZn3slCWulg/#scalars&_smoothingWeight=0)|
| 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) [TB.dev](https://tensorboard.dev/experiment/n2UN83TkTdyKZn3slCWulg/#scalars&_smoothingWeight=0)| | 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) \| [TB.dev](https://tensorboard.dev/experiment/n2UN83TkTdyKZn3slCWulg/#scalars&_smoothingWeight=0)|
| 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) [TB.dev](https://tensorboard.dev/experiment/n2UN83TkTdyKZn3slCWulg/#scalars&_smoothingWeight=0)| | 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) \| [TB.dev](https://tensorboard.dev/experiment/n2UN83TkTdyKZn3slCWulg/#scalars&_smoothingWeight=0)|
#### Mobile-size RetinaNet (Trained from scratch):
backbone | resolution | epochs | FLOPs (B) | params (M) | box AP | download
------------ | :--------: | -----: | --------: | ---------: | -----: | -------:
Mobile SpineNet-49 | 384x384 | 600 | 1.0 | 2.32 | 28.1 | [config](https://github.com/tensorflow/models/blob/master/official/vision/beta/configs/experiments/retinanet/coco_spinenet49_mobile_tpu.yaml) |
### Instance Segmentation Baselines ### Instance Segmentation Baselines
......
official/vision/beta/configs/experiments/retinanet/coco_spinenet49_mobile_tpu.yaml
View file @ 1cb5e572
# --experiment_type=retinanet_spinenet_mobile_coco
runtime: runtime:
distribution_strategy: 'tpu' distribution_strategy: 'tpu'
mixed_precision_dtype: 'bfloat16' mixed_precision_dtype: 'bfloat16'
...@@ -36,7 +37,7 @@ task: ...@@ -36,7 +37,7 @@ task:
parser: parser:
aug_rand_hflip: true aug_rand_hflip: true
aug_scale_max: 2.0 aug_scale_max: 2.0
aug_scale_min: 0.1 aug_scale_min: 0.5
validation_data: validation_data:
dtype: 'bfloat16' dtype: 'bfloat16'
global_batch_size: 8 global_batch_size: 8
......
official/vision/beta/configs/experiments/retinanet/coco_spinenet49s_mobile_tpu.yaml
View file @ 1cb5e572
# --experiment_type=retinanet_spinenet_mobile_coco
runtime: runtime:
distribution_strategy: 'tpu' distribution_strategy: 'tpu'
mixed_precision_dtype: 'bfloat16' mixed_precision_dtype: 'bfloat16'
...@@ -36,7 +37,7 @@ task: ...@@ -36,7 +37,7 @@ task:
parser: parser:
aug_rand_hflip: true aug_rand_hflip: true
aug_scale_max: 2.0 aug_scale_max: 2.0
aug_scale_min: 0.1 aug_scale_min: 0.5
validation_data: validation_data:
dtype: 'bfloat16' dtype: 'bfloat16'
global_batch_size: 8 global_batch_size: 8
......
official/vision/beta/configs/experiments/retinanet/coco_spinenet49xs_mobile_tpu.yaml
View file @ 1cb5e572
# --experiment_type=retinanet_spinenet_mobile_coco
runtime: runtime:
distribution_strategy: 'tpu' distribution_strategy: 'tpu'
mixed_precision_dtype: 'bfloat16' mixed_precision_dtype: 'bfloat16'
...@@ -36,7 +37,7 @@ task: ...@@ -36,7 +37,7 @@ task:
parser: parser:
aug_rand_hflip: true aug_rand_hflip: true
aug_scale_max: 2.0 aug_scale_max: 2.0
aug_scale_min: 0.1 aug_scale_min: 0.5
validation_data: validation_data:
dtype: 'bfloat16' dtype: 'bfloat16'
global_batch_size: 8 global_batch_size: 8
......
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