Использование tenorflow hub с go

Question

Я хочу использовать предварительно обученные модели в своем приложении go.Особенно модель Inception-ResNet-v2.Похоже, что эта модель доступна только через концентратор тензорного потока (https://www.tensorflow.org/hub/).

. Однако я не смог найти никакой документации по использованию концентратора тензорного потока с привязками языка go для тензорного потока.

Какя могу скачать и использовать эти модели на ходу?

Answer 1

Итак, после большой работы за последние несколько дней я наконец нашел способ.

Сначала я хотел просто использовать Python, чтобы выполнить всю работу с Tensorflow, а затем предоставить результаты через службу отдыха. Однако оказалось, что количество моделей, предоставляемых Tensorflow Hub, очень мало. Это было проблемой для меня, потому что мне приходилось пробовать разные модели и сравнивать их.

Таким образом, я перешел на использование моделей с https://github.com/tensorflow/models. Существует несколько руководств по экспорту данных в файлы .pb. Эти файлы затем могут быть загружены в Go с помощью gocv.

Требуется много работы для конвертации файлов, но в конце я думаю, что это лучший способ использовать модели Tensorflow на ходу.