Любой код для преобразования модели тензорного потока в модель кофе?

Question

Моя конечная цель - использовать TensorRT для оптимизации моей модели для развертывания. Я делаю свои эксперименты в тензорном потоке. Но TensorRT требует модель ввода в формате caffemodel.

Answer 1

В новый TensorRT 3 добавлена ​​поддержка TensorFlow.

Answer 2

Начиная с TensorRT 2.1 поддержка преобразования модели TensorFlow отсутствует. Это может быть поддержано в будущем.

Возможно, вы захотите изучить варианты преобразования модели TensorFlow в Caffe, а затем преобразовать ее с помощью TensorRT в механизм вывода.