Как преобразовать контрольные точки в save_model_pb? В тензорном потоке 2.x

#tensorflow #tensorflow2.0 #object-detection #s

Вопрос:

Я обучил модель Retinanet, используя следующее:

https://github.com/tensorflow/models/tree/master/official/vision/detection

Я получаю контрольные точки следующим образом:

 ctl_step_5000.ckpt-52.data-00000-of-00001
ctl_step_5000.ckpt-52.index
 

Как я могу преобразовать это в файл saved_model.pb?

Я использую tf2.3

Комментарии:

1. в файлах контрольных точек хранятся только данные, а не архитектура модели. Вам это понадобится в виде кода для загрузки контрольной точки. model.load_weights(путь к контрольной точке)

2. Не могли бы вы поподробнее, пожалуйста?

3. Возможно, вам потребуется сохранить всю модель целиком, чтобы сохранить ее в saved_model.pb формате. Архитектура модели и конфигурация обучения (включая оптимизатор, потери и показатели) хранятся в saved_model.pb . Проверьте эту ссылку для получения более подробной информации.