Как конвертировать модели 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