Как конвертировать модели Niftynet в формат .pb

Question

Я бы хотел использовать модели в зоопарке Niftynet в tenorflowjs.

Когда я загружаю модели из зоопарка Niftynet, они имеют формат контрольных точек (ckpt). Однако tenorflowjs ожидает, что модели будут в формате pb. Кроме того, формат контрольных точек не содержит графической информации, поэтому нам нужно сохранить модель, используя simple_save или аналогичный API [1].

Итак, что я собираюсь сделать, это сделать:

1. Скачать модель из зоопарка Niftynet
2. Восстановите это. (на этом этапе мне нужна помощь)
3. Сохраните всю модель, используя simple_save api. Это приведет к модели .pb. (на этом этапе мне нужна помощь)
4. Используйте tenorflowjs-converter [2], чтобы преобразовать его в формат pb, который может использоваться tenorflowjs.

Так может ли кто-нибудь из сообщества niftynet помочь мне с шагами 2 и 3: как загрузить модель, используя файлы контрольных точек и файл конфигурации, и полностью сохранить ее с помощью API simple_save.

1. https://www.tensorflow.org/guide/saved_model
2. https://js.tensorflow.org/tutorials/import-saved-model.html