File tree Expand file tree Collapse file tree 1 file changed +8
-3
lines changed
research/object_detection/g3doc Expand file tree Collapse file tree 1 file changed +8
-3
lines changed Original file line number Diff line number Diff line change 3
3
After your model has been trained, you should export it to a Tensorflow
4
4
graph proto. A checkpoint will typically consist of three files:
5
5
6
- * model.ckpt-${CHECKPOINT_NUMBER}.data-00000-of-00001,
6
+ * model.ckpt-${CHECKPOINT_NUMBER}.data-00000-of-00001
7
7
* model.ckpt-${CHECKPOINT_NUMBER}.index
8
8
* model.ckpt-${CHECKPOINT_NUMBER}.meta
9
9
@@ -16,7 +16,12 @@ python object_detection/export_inference_graph.py \
16
16
--input_type image_tensor \
17
17
--pipeline_config_path ${PIPELINE_CONFIG_PATH} \
18
18
--trained_checkpoint_prefix ${TRAIN_PATH} \
19
- --output_directory output_inference_graph.pb
19
+ --output_directory ${EXPORT_DIR}
20
20
```
21
21
22
- Afterwards, you should see a graph named output_inference_graph.pb.
22
+ Afterwards, you should see the directory ${EXPORT_DIR} containing the following:
23
+
24
+ * output_inference_graph.pb, the frozen graph format of the exported model
25
+ * saved_model/, a directory containing the saved model format of the exported model
26
+ * model.ckpt.* , the model checkpoints used for exporting
27
+ * checkpoint, a file specifying to restore included checkpoint files
You can’t perform that action at this time.
0 commit comments