Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
ResNet50_tensorflow
Commits
1cb5e572
Commit
1cb5e572
authored
Apr 06, 2021
by
Xianzhi Du
Committed by
A. Unique TensorFlower
Apr 06, 2021
Browse files
Internal change
PiperOrigin-RevId: 367083514
parent
9fe85228
Changes
4
Show whitespace changes
Inline
Side-by-side
Showing
4 changed files
with
15 additions
and
6 deletions
+15
-6
official/vision/beta/MODEL_GARDEN.md
official/vision/beta/MODEL_GARDEN.md
+9
-3
official/vision/beta/configs/experiments/retinanet/coco_spinenet49_mobile_tpu.yaml
...igs/experiments/retinanet/coco_spinenet49_mobile_tpu.yaml
+2
-1
official/vision/beta/configs/experiments/retinanet/coco_spinenet49s_mobile_tpu.yaml
...gs/experiments/retinanet/coco_spinenet49s_mobile_tpu.yaml
+2
-1
official/vision/beta/configs/experiments/retinanet/coco_spinenet49xs_mobile_tpu.yaml
...s/experiments/retinanet/coco_spinenet49xs_mobile_tpu.yaml
+2
-1
No files found.
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
| 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-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-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-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
...
...
official/vision/beta/configs/experiments/retinanet/coco_spinenet49_mobile_tpu.yaml
View file @
1cb5e572
# --experiment_type=retinanet_spinenet_mobile_coco
runtime
:
distribution_strategy
:
'
tpu'
mixed_precision_dtype
:
'
bfloat16'
...
...
@@ -36,7 +37,7 @@ task:
parser
:
aug_rand_hflip
:
true
aug_scale_max
:
2.0
aug_scale_min
:
0.
1
aug_scale_min
:
0.
5
validation_data
:
dtype
:
'
bfloat16'
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
:
distribution_strategy
:
'
tpu'
mixed_precision_dtype
:
'
bfloat16'
...
...
@@ -36,7 +37,7 @@ task:
parser
:
aug_rand_hflip
:
true
aug_scale_max
:
2.0
aug_scale_min
:
0.
1
aug_scale_min
:
0.
5
validation_data
:
dtype
:
'
bfloat16'
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
:
distribution_strategy
:
'
tpu'
mixed_precision_dtype
:
'
bfloat16'
...
...
@@ -36,7 +37,7 @@ task:
parser
:
aug_rand_hflip
:
true
aug_scale_max
:
2.0
aug_scale_min
:
0.
1
aug_scale_min
:
0.
5
validation_data
:
dtype
:
'
bfloat16'
global_batch_size
:
8
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment