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
Show 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).
...
@@ -19,9 +19,14 @@ uses a ResNet50 model implemented in [`resnet_model.py`](./resnet_model.py).
To use
To use
either dataset, make sure that you have the latest version of TensorFlow
either dataset, make sure that you have the latest version of TensorFlow
installed and
installed and
[
add the models folder to your Python path
](
/official/#running-the-models
)
,
[
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`
.
### 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
### CIFAR-10
...
@@ -43,7 +48,7 @@ location with the `--data_dir` flag, like:
...
@@ -43,7 +48,7 @@ location with the `--data_dir` flag, like:
python resnet_cifar_main.py
--data_dir
=
/path/to/cifar
python resnet_cifar_main.py
--data_dir
=
/path/to/cifar
```
```
### ImageNet
### ImageNet
Training
Download the ImageNet dataset and convert it to TFRecord format.
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
)
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