Skip to content

Instantly share code, notes, and snippets.

@nvnnghia
Last active April 10, 2018 00:47
Show Gist options
  • Save nvnnghia/e84940f022244591c674bfe9864fff1d to your computer and use it in GitHub Desktop.
Save nvnnghia/e84940f022244591c674bfe9864fff1d to your computer and use it in GitHub Desktop.
# Source:
https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/running_pets.md
# Create data:
python object_detection/dataset_tools/create_pascal_tf_record.py --data=/home/phatv/nghia/mxnet-yolo/data1/VOCdevkit/ --year=VOC2012 --output_path=val.record --set=val
# Train:
python train.py --logtostderr --train_dir=train/ --pipeline_config_path=train/ssd_mobilenet_v11_coco.config
#combine ckpt to inference graph:
python export_inference_graph.py --input_type image_tensor --pipeline_config_path train/ssd_mobilenet_v11_coco.config --trained_checkpoint_prefix train/model.ckpt-85863 --output_directory exported_graphs1
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment