我正在关注 Dat Trans使用 TensorFlow 的对象检测器 API 训练我自己的对象检测器的示例。
我成功开始训练自定义对象。我正在使用 CPU 来训练模型，但完成 100 个训练步骤大约需要 3 个小时。我想我必须更改 .config 中的一些参数。
我尝试将 .ckpt 转换为 .pb，我引用了 this发布，但我仍然无法转换
1)如何减少训练步数？
2) 有没有办法将.ckpt 转换为.pb。
请您参考如下方法:
我不认为你可以减少训练步数，但你可以在任何检查点(ckpt)停止，然后将其转换为.pb文件< br/> 从 TensorFlow 模型 git 存储库中，您可以使用，export_inference_graph.py
和下面的代码
python tensorflow_models/object_detection/export_inference_graph.py \
--input_type image_tensor \
--pipeline_config_path architecture_used_while_training.config \
--trained path_to_saved_ckpt/model.ckpt-NUMBER \
--output_directory model/
其中 NUMBER 指的是您最新保存的检查点文件编号，但是如果您在 tensorboard 中发现它更好，您可以使用旧的检查点文件
