Как загрузить контрольную точку и вывод с C ++ для тензорного потока?
У меня есть Tensorflow с Python API и получил файл модели контрольной точки:
model.ckpt-17763.data-00000-of-00001
model.ckpt-17763.index
model.ckpt-17763.meta
Но я хочу использовать совместно используемую библиотеку C / C ++ (.so файл) при интеграции в производство. Поэтому мне нужно загрузить этот файл модели и логический вывод с кодом C ++ и скомпилировать в общую библиотеку. Есть ли учебник или пример для этого?
Решение
Вы можете написать код C ++ для загрузки и использовать свой график с инструкциями Вот.
Вы можете использовать файлы Вот сделать проект Cmake с тензорным потоком вне репозитория TF и скомпилировать вашу библиотеку.
Тем не менее, вам все еще нужны файлы .ckpt рядом с вашим .so, я не знаю, как интегрировать их в него.
Есть много вопросов об этом на S.O., и несколько учебных пособий (см. Два, процитированных в этот ответ) но так как тензорный поток быстро развивается, они быстро устаревают, и всегда нужно немного потрудиться, чтобы заставить его работать должным образом (полностью выполнимо и, тем не менее, становится легче).
Другие решения
Других решений пока нет …
detector