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
61aee86e
Commit
61aee86e
authored
Nov 13, 2019
by
Hongkun Yu
Committed by
A. Unique TensorFlower
Nov 13, 2019
Browse files
Add pretrained checkpoints and TFHub links
PiperOrigin-RevId: 280321302
parent
8475c0bb
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
9 additions
and
4 deletions
+9
-4
official/vision/image_classification/README.md
official/vision/image_classification/README.md
+9
-4
No files found.
official/vision/image_classification/README.md
View file @
61aee86e
...
...
@@ -19,9 +19,14 @@ uses a ResNet50 model implemented in [`resnet_model.py`](./resnet_model.py).
To use
either dataset, make sure that you have the latest version of TensorFlow
installed and
[
add the models folder to your Python path
](
/official/#running-the-models
)
,
otherwise you may encounter an error like
`ImportError: No module named
official.resnet`
.
[
add the models folder to your Python path
](
/official/#running-the-models
)
.
### Pretrained Models
*
[
ResNet50 Checkpoints
](
https://storage.googleapis.com/cloud-tpu-checkpoints/resnet/resnet50.tar.gz
)
*
ResNet50 TFHub:
[
feature vector
](
https://tfhub.dev/tensorflow/resnet_50/feature_vector/1
)
and
[
classification
](
https://tfhub.dev/tensorflow/resnet_50/classification/1
)
### CIFAR-10
...
...
@@ -43,7 +48,7 @@ location with the `--data_dir` flag, like:
python resnet_cifar_main.py
--data_dir
=
/path/to/cifar
```
### ImageNet
### ImageNet
Training
Download the ImageNet dataset and convert it to TFRecord format.
The following
[
script
](
https://github.com/tensorflow/tpu/blob/master/tools/datasets/imagenet_to_gcs.py
)
...
...
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