tensorflow训练模型配置文件详解

参考官方:Configuring the Object Detection Training Pipeline

Tensorflow目标检测API使用protobuf文件来配置训练和测试相关的东西。

文件格式可以参照object_detection/protos/pipeline.proto文件:

  1. model_config。这里定义训练模型类型和使用参数。
    model_config.proto

  1. train_config,定义哪些参数应该被用来训练模型参数。
    train_config.proto

  1. eval_config,设定的使用哪些指标将被报告进行评估。

  1. train_input_config,定义在哪些数据集是训练集。
  2. eval_input_config,定义在哪些数据集进行评估的。通常,这应该与训练输入数据集不同。

配置文件分为5个部分:

 

发表评论