Как загрузить и предсказать предварительно обученную модель тензорного потока в код Java?

Question

Я не уверен, какова обычная практика загрузки предварительно обученной модели keras в код Java?

Я видел deeplearning4j и Поддержка нативной Java от Google .

Удивительно, но Google говорит:

Внимание: API-интерфейс TensorFlow Java не покрывается гарантиями стабильности TensorFlow API.

Я в замешательстве, что здесь происходит, и какие соображения мне нужно принять во внимание, прежде чем выбрать движок Java с тензорным потоком.Очевидно, я хотел бы поддержать последнюю версию CUDA / cuDNN.Мое приложение работает на ПК, это не мобильное приложение.

Answer 1

Вы можете выполнить следующие действия:

1. Вы можете использовать TensorFlow Lite или TensorFlow Mobile.Рекомендуется TensorFlow Lite.Посетите здесь .
2. TensorFlow Lite имеет зависимость gradle, а также maven, что полезно в Android и java.
3. Для преобразования вашей модели python в файл TF Lite используйте функциив этой записной книжке
4. Он может выполнять умозаключения для модели, которую вы обучали на python.Но в TF Lite есть проблемы с ячейками LSTM и слоями BatchNormalization.