Skip to content

Commit e7ecb9f

Browse files
authored
Merge pull request #10039 from rmlarsen/branch_155393864
Branch 155393864
2 parents 38c1b53 + 5c3425c commit e7ecb9f

File tree

2 files changed

+14
-6
lines changed

2 files changed

+14
-6
lines changed

performance/performance_models.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -345,7 +345,7 @@ executing the main script
345345
* **`num_gpus`**: Number of GPUs to use.
346346
* **`data_dir`**: Path to data to process. If not set, synthetic data is used.
347347
To use Imagenet data use these
348-
[instructions(https://github.com/tensorflow/models/tree/master/inception#getting-started)
348+
[instructions](https://github.com/tensorflow/models/tree/master/inception#getting-started)
349349
as a starting point.
350350
* **`batch_size`**: Batch size for each GPU.
351351
* **`variable_update`**: The method for managing variables: `parameter_server`

tutorials/image_retraining.md

Lines changed: 13 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -14,11 +14,16 @@ laptop, without requiring a GPU. This tutorial will show you how to run the
1414
example script on your own images, and will explain some of the options you have
1515
to help control the training process.
1616

17+
Note: This version of the tutorial mainly uses bazel. A bazel free version is
18+
also available
19+
[as a codelab](https://codelabs.developers.google.com/codelabs/tensorflow-for-poets/#0).
20+
1721
[TOC]
1822

1923
## Training on Flowers
2024

2125
![Daisies by Kelly Sikkema](https://www.tensorflow.org/images/daisies.jpg)
26+
2227
[Image by Kelly Sikkema](https://www.flickr.com/photos/95072945@N05/9922116524/)
2328

2429
Before you start any training, you'll need a set of images to teach the network
@@ -148,10 +153,10 @@ Here's an example of how to build and run the label_image example with your
148153
retrained graphs:
149154

150155
```sh
151-
bazel build tensorflow/examples/label_image:label_image && \
152-
bazel-bin/tensorflow/examples/label_image/label_image \
156+
bazel build tensorflow/examples/image_retraining:label_image && \
157+
bazel-bin/tensorflow/examples/image_retraining/label_image \
153158
--graph=/tmp/output_graph.pb --labels=/tmp/output_labels.txt \
154-
--output_layer=final_result \
159+
--output_layer=final_result:0 \
155160
--image=$HOME/flower_photos/daisy/21652746_cc379e0eea_m.jpg
156161
```
157162

@@ -160,7 +165,10 @@ You should see a list of flower labels, in most cases with daisy on top
160165
`--image` parameter with your own images to try those out, and use the C++ code
161166
as a template to integrate with your own applications.
162167

163-
If you'd like to use the retrained model in a Python program [this example from @eldor4do shows what you'll need to do](https://github.com/eldor4do/TensorFlow-Examples/blob/master/retraining-example.py).
168+
If you'd like to use the retrained model in your own Python program, then the
169+
above
170+
[`label_image` script](https://www.tensorflow.org/code/tensorflow/examples/image_retraining/label_image.py)
171+
is a reasonable starting point.
164172

165173
## Training on Your Own Categories
166174

@@ -260,7 +268,7 @@ them destroys their meaning.
260268

261269
There are several other parameters you can try adjusting to see if they help
262270
your results. The `--learning_rate` controls the magnitude of the updates to the
263-
final layer during training. Intuitively if this is smaller then the learning
271+
final layer during training. Intuitively if this is smaller than the learning
264272
will take longer, but it can end up helping the overall precision. That's not
265273
always the case though, so you need to experiment carefully to see what works
266274
for your case. The `--train_batch_size` controls how many images are examined

0 commit comments

Comments
 (0)