Тензор потока, измените модель контрольной точки (.meta .index .data) на замороженную модель (.pd)

Question

Я не знаком с тензорным потоком.

Я хочу преобразовать эту сеть, https://github.com/jiangsutx/SRN-Deblur, из тензорного потока в nvidia tenorRT. Ему нужен файл модели ".pb", но проект дает только три файла модели следующим образом:

• deblur.model-52300.data-00000-оф-00001
• deblur.model-52300.index
• deblur.model-52300.meta

Поэтому я хочу преобразовать эти файлы в файл ".pb".

Я проверил идеи, данные:

• https://blog.metaflow.fr/tensorflow-how-to-freeze-a-model-and-serve-it-with-a-python-api-d4f3596b3adc
• Тензор потока: как преобразовать файлы моделей.meta, .data и.index в один файл graph.pb

Проблема заключается в том, что обе приведенные идеи не удаются из-за того, что get_checkpoint_state() и latest_checkpoint() дают значение None.

Это вызвано отсутствием файла контрольной точки?

Есть ли другие способы реализовать это?

Любая идея приветствуется.

Благодарю.

Answer 1

Как видно из собственного репо: они используют get_checkpoint_state протестировать предварительно обученную модель. https://github.com/jiangsutx/SRN-Deblur/blob/master/models/model.py

Поэтому я бы сказал, что да, это из-за отсутствия файла.ckpt, не предоставленного автором.

По опыту, обычно первый метод из metaflow работает довольно хорошо.