Skip to content

Commit e4770fb

Browse files
authored
Update walkthrough-tensorflow-object-detection.md
1 parent 902f286 commit e4770fb

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

examples/walkthrough-tensorflow-object-detection.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -118,15 +118,15 @@ The model checkpoints and outputs for this task will save in the `training` fold
118118
<img src="images/tf_sc.jpg" width="500" />
119119
</p>
120120

121-
We ran this model for about **10,000 steps**, and it took **37 hours** (local CPU). If you want to run a faster model, we recommend trying `ssd_mobilenet_v1_coco` in TensorFlow model zoo, training on a GPU, or using Google's [Cloud Machine Learning Engine](https://cloud.google.com/ml-engine/).
121+
We ran this model for about **20,000 steps**, and it took **50 hours** (local CPU). If you want to run a faster model, we recommend trying `ssd_mobilenet_v1_coco` in TensorFlow model zoo, training on a GPU, or using Google's [Cloud Machine Learning Engine](https://cloud.google.com/ml-engine/).
122122

123123
## Visualize the Model
124124
Create the building detection model inference graph with:
125125

126126
```shell
127127
python export_inference_graph.py --input_type image_tensor \
128128
--pipeline_config_path training/ssd_inception_v2_coco.config \
129-
--trained_checkpoint_prefix training/model.ckpt-9575
129+
--trained_checkpoint_prefix training/model.ckpt-18699 \
130130
--output_directory building_od_ssd
131131
```
132132
We can visualize this graph using [`tensorboard`](https://github.com/tensorflow/tensorboard):

0 commit comments

Comments
 (0)