终于到了能动手实践的阶段啦。依例先读一下文档
Running Locally
准备工作是否就绪了呢?
- 安装妥当否
- data created 否
- 配置妥当否
建议的文件结构
- data
- label_map file
- train TFRecord file
- eval TFRecord file
- models
- model
- pipeline config file
- train
- eval
- model
跑一次训练吧
# From the tensorflow/models/research/ directory
python object_detection/train.py \
--logtostderr \
--pipeline_config_path=${PATH_TO_YOUR_PIPELINE_CONFIG} \
--train_dir=${PATH_TO_TRAIN_DIR}
这里有个坑,文档里也提醒到了,就是如果不人为中断的话,训练会一直一直地持续下去。
是不是迫不及待地想试试呢