Tensorflow对象检测配置文件文档

Question

我在tensorflow中使用对象检测api.我注意到几乎所有参数都通过配置文件.我找不到关于这些配置文件的选项的任何文档或教程.

我知道在官方的git中,他们为他们的预训练模型提供了一个配置文件列表,这可能非常有用,但它并不涵盖所有情况,当然如果需要也不提供任何解释.

例如,在train_config一节中有一些数据扩充选项是非常自我解释但其他选项的潜在存在不清楚:

  data_augmentation_options {
    random_horizontal_flip {
    }
  }
  data_augmentation_options {
    ssd_random_crop {
    }
  }

有没有我可以参考的来源？例如,在本教程中,出现了两个我不知道的额外选项(batch_queue_capacity和prefetch_queue_capacity).我在哪里可以找到一个合适的选项列表？我知道它是特定于模型的,但其中一些是通用的,非常有用.

Answer 1

正如所提到的配置文档,配置文件只是协议缓冲器中所描述的对象.proto下的文件research/object_detection/protos.顶级对象是TrainEvalPipelineConfig定义的pipeline.proto,不同的文件描述每个元素.例如,data_augmentation_options是PreprocessingStep定义的对象preprocessor.proto(反过来可以包括一系列用于不同预处理任务的其他可能对象).每个对象和字段的含义可能是也可能不是很明显或没有详细记录,但您始终可以参考源代码以确切了解每个值的使用方式(例如,检查preprocessor.py以了解如何完成数据扩充).