- Я сохранил модель keras с весами с python и загрузил
ComputationGraph в java на DL4j, используя
model = KerasModelImport.importKerasModelAndWeights(unet, enforceTrainingConfig=false)
Я создаю свой ввод INArray с помощью Nd4j.create(floats) и запускаю логический вывод:
INDArray output = model.output(input)
Я получаю вывод:
float[][] x = output.reshape(new int[]{floats.length, 256 * 256}).toFloatMatrix();
Выход есть, но похож на вход с другим шаблоном. Результат не верный.
Я попробовал эту модель, она хорошо работает в python с Keras и в Java с frozenModel (freezegraph) на Tensorflow. Кто-нибудь пытается использовать обученную модель керас на dl4j?
Я ожидаю воспроизвести вывод на dl4j, как я получил на Tensorflow.
Вы когда-нибудь сталкивались с этой проблемой? Любые советы будут полезны, спасибо.
ввод
вывод