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
f4baddb3
Commit
f4baddb3
authored
May 29, 2020
by
Pengchong Jin
Committed by
A. Unique TensorFlower
May 29, 2020
Browse files
Update README.
PiperOrigin-RevId: 313806578
parent
72587577
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
7 additions
and
1 deletion
+7
-1
official/vision/detection/README.md
official/vision/detection/README.md
+7
-1
No files found.
official/vision/detection/README.md
View file @
f4baddb3
...
@@ -41,7 +41,7 @@ python3 ~/models/official/vision/detection/main.py \
...
@@ -41,7 +41,7 @@ python3 ~/models/official/vision/detection/main.py \
The pre-trained ResNet-50 checkpoint can be found here:
The pre-trained ResNet-50 checkpoint can be found here:
```
```
gs://cloud-tpu-artifacts/resnet
/resnet
-nhwc
-2018-
10-14/model.ckpt-112602
https://storage.cloud.google.com/cloud-tpu-checkpoints/model-garden-vision/detection
/resnet
50
-2018-
02-07.tar.gz
```
```
### Train a custom RetinaNet using the config file.
### Train a custom RetinaNet using the config file.
...
@@ -150,6 +150,12 @@ python3 ~/models/official/vision/detection/main.py \
...
@@ -150,6 +150,12 @@ python3 ~/models/official/vision/detection/main.py \
--params_override
=
"{train: { checkpoint: { path:
${
RESNET_CHECKPOINT
}
, prefix: resnet50/ }, train_file_pattern:
${
TRAIN_FILE_PATTERN
}
}, eval: { val_json_file:
${
VAL_JSON_FILE
}
, eval_file_pattern:
${
EVAL_FILE_PATTERN
}
} }"
--params_override
=
"{train: { checkpoint: { path:
${
RESNET_CHECKPOINT
}
, prefix: resnet50/ }, train_file_pattern:
${
TRAIN_FILE_PATTERN
}
}, eval: { val_json_file:
${
VAL_JSON_FILE
}
, eval_file_pattern:
${
EVAL_FILE_PATTERN
}
} }"
```
```
The pre-trained ResNet-50 checkpoint can be found here:
```
https://storage.cloud.google.com/cloud-tpu-checkpoints/model-garden-vision/detection/resnet50-2018-02-07.tar.gz
```
### Train a custom Mask R-CNN using the config file.
### Train a custom Mask R-CNN using the config file.
First, create a YAML config file, e.g.
*my_maskrcnn.yaml*
.
First, create a YAML config file, e.g.
*my_maskrcnn.yaml*
.
...
...
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