Понимание export_tflite_ssd_graph.py - PullRequest
Новая
       27

Понимание export_tflite_ssd_graph.py

0 голосов
/ 18 марта 2019

Здесь - руководство по преобразованию Mobilenet + SSD в tflite, в какой-то момент они используют export_tflite_ssd_graph.py , насколько я понимаю, этот пользовательский скрипт используется для поддержки операции tf.image.non_max_suppression. 

export CONFIG_FILE=gs://${YOUR_GCS_BUCKET}/data/pipeline.config
export CHECKPOINT_PATH=gs://${YOUR_GCS_BUCKET}/train/model.ckpt-2000
export OUTPUT_DIR=/tmp/tflite

python object_detection/export_tflite_ssd_graph.py \
--pipeline_config_path=$CONFIG_FILE \
--trained_checkpoint_prefix=$CHECKPOINT_PATH \
--output_directory=$OUTPUT_DIR \
--add_postprocessing_op=true

Но мне интересно, что такое pipeline.config и как его создать, если я использую пользовательскую модель (например, FaceBoxes ), которая использует операцию tf.image.non_max_suppression?

1 Ответ

1 голос
/ 18 марта 2019

Основная цель export_tflite_ssd_graph.py состоит в том, чтобы экспортировать файлы контрольных точек обучения в замороженный график, который впоследствии можно будет использовать для обучения переноса или для прямого вывода (поскольку они содержат информацию о структуре модели, а также информацию об обученных весах).Фактически, все модели, перечисленные в модель зоопарка , представляют собой замороженный график, сгенерированный таким образом.

Что касается tf.image.non_max_suppression, то export_tflite_ssd_graph.py не используется для его «поддержки», но если --add_postprocessing_op установлен true, к замороженному графу будет добавлен другой пользовательский операционный узел, этот пользовательский узелбудет иметь функциональность, аналогичную op tf.image.non_max_suppression.См. Ссылку здесь .

Наконец, файл pipeline.config напрямую соответствует файлу конфигурации, который вы используете для обучения (--pipeline_config_path), он является его копией, но часто сизмененный порог оценки (см. описание здесь о pipeline.config.), поэтому вам придется создать его до начала тренировки, если вы используете пользовательскую модель.А для создания пользовательского файла конфигурации здесь является официальным руководством.

...