User Tools
Differences
This shows you the differences between two versions of the page.
| Both sides previous revision Previous revision | Next revision Both sides next revision | ||
|
cs:vision:object_detection:start [2018/03/31 18:16] Mike Bykhovtsev Added instructions on how to convert trained data |
cs:vision:object_detection:start [2018/03/31 19:46] James Irwin [Getting a Model] |
||
|---|---|---|---|
| Line 95: | Line 95: | ||
| ===== Getting a Model ===== | ===== Getting a Model ===== | ||
| Instructions were adapted from [[https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/configuring_jobs.md | here]]. | Instructions were adapted from [[https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/configuring_jobs.md | here]]. | ||
| - | One of the advantages of using TFODA is that it is really easy to try different network architectures (models) and seeing their speed vs. accuracy tradeoffs. To get a model | + | One of the advantages of using TFODA is that it is really easy to try different network architectures (models) and seeing their speed vs. accuracy tradeoffs. Example config files can be found [[https://github.com/tensorflow/models/tree/master/research/object_detection/samples/configs : here]]. You'll need to modify these config files a bit for your own use. In addition, most of the models have network weights that have been pretrained on some dataset. Starting from these check-points is usually much faster than training from scratch. You can locate them [[https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/detection_model_zoo.md | here]]. |
| + | |||
| + | Place the model config file and the pretrained model in the models/ directory of your workspace. | ||
| + | |||
| + | ===== Start Training ===== | ||
| + | The following command assume your current working directory is the root of the workspace you created earlier. To start training, run: | ||
| + | python ~/.local/tensorflow_object_detection_api/research/object_detection/train.py \ | ||
| + | --logtostderr \ | ||
| + | --pipeline_config_path=<model config file> \ | ||
| + | --train_dir= | ||
| ===== Exporting a trained model for inference ===== | ===== Exporting a trained model for inference ===== | ||
| To export checkpoint trained data for ''%%robosub_object_detection%%'' format you need to follow [[https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/exporting_models.md|these]] instructions. Or run this: | To export checkpoint trained data for ''%%robosub_object_detection%%'' format you need to follow [[https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/exporting_models.md|these]] instructions. Or run this: | ||
