Вывод по TPU для модели, обученной на GPU (API Tensorflow Object Detection)

Я пытаюсь следовать приведенному ниже руководству по экспорту некоторой модели обнаружения объектов (на основе API обнаружения объектов Tensorflow), обученной с помощью графического процессора для использования в TPU для вывода,

https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/tpu_exporters.md

  1. В одном из требований упоминалось:
    «Предполагается, что у пользователей есть: PIPELINE_CONFIG: файл конфигурации pipe_pb2.TrainEvalPipelineConfig», но я не могу найти файл pipe_pb2.TrainEvalPipelineConfig где-либо в Интернете или в любом репозитории, могу ли я знать как получить файл?

  2. Что такое «INPUT_PLACEHOLDER: имя заполнителя ввода в signal_def_map модели», где я могу его найти?

  3. Что такое «INPUT_TYPE: тип входного узла, который может быть одним из «image_tensor», «encoded_image_string_tensor» или «tf_example»? Где я могу найти его?

  4. Где я могу получить пример, связанный с «выполнением вывода на TPU с использованием модели обнаружения объектов, обученной на GPU»?

С уважением, Чу Кок Ва


google-cloud-tpu
person Chew Kok Wah    schedule 05.03.2020    source источник

Ответы (1)


arrow_upward
1
arrow_downward

  1. Прототип определен по адресу: https://github.com/tensorflow/models/blob/2dfd1e6388c06eb47945f2592d5dad7609172491/research/object_detection/protos/pipeline.proto#L14 Предполагается, что вы используете ту же конфигурацию, которую вы использовали при обучении.

  2. Он находится в MetaGraphDef в экспортированной модели SavedModel. SavedModel: https://github.com/tensorflow/tensorflow/blob/master/tensorflow/core/protobuf/saved_model.proto MetaGraphDef.signature_def: https://github.com/tensorflow/tensorflow/blob/0bf070c54d015ee86de8328d8bdb5582dc0f6d93/tensorflow/core/protobuf/meta_graph.proto#L89

  3. Это зависит от того, какой тип данных вы хотите передать. Определено на https://github.com/tensorflow/models/blob/2dfd1e6388c06eb47945f2592d5dad7609172491/research/object_detection/exporter.py#L199

  4. Пример можно найти по адресу https://github.com/tensorflow/models/blob/2dfd1e6388c06eb47945f2592d5dad7609172491/research/object_detection/tpu_exporters/export_saved_model_tpu_lib_test.py#L56 Это в основном экспорт сохраненной модели TPU с учетом контрольной точки и конфигурации конвейера.

person Zhuoran Liu    schedule 12.06.2020