Как исправить jnius.JavaException при использовании Pyjnius - PullRequest
Новая
       19

Как исправить jnius.JavaException при использовании Pyjnius

1 голос
/ 04 июня 2019

Я пытаюсь получить этот пример для импорта импорта TensorFlow в Java для запуска: https://github.com/deeplearning4j/tf-import

Мне удалось заставить импорт Java в Mnist.java работать после некоторых модификаций. Однако я не могу заставить mnist_jumpy.py работать, чтобы использовать модель в DeepLearning4J из Python. Я запустил его с изменениями ниже, но я получаю это исключение при загрузке модели: 

log4j:WARN No appenders could be found for logger (org.nd4j.linalg.factory.Nd4jBackend).
log4j:WARN Please initialize the log4j system properly.
log4j:WARN See http://logging.apache.org/log4j/1.2/faq.html#noconfig for more info.
WARNING: An illegal reflective access operation has occurred
WARNING: Illegal reflective access by com.rits.cloning.Cloner (file:/home/micha/.deeplearning4j/pydl4j/pydl4j-1.0.0-SNAPSHOT-cpu-core-datavec-spark2-2.11/pydl4j-1.0.0-SNAPSHOT-bin.jar) to field java.util.TreeSet.m
WARNING: Please consider reporting this to the maintainers of com.rits.cloning.Cloner
WARNING: Use --illegal-access=warn to enable warnings of further illegal reflective access operations
WARNING: All illegal access operations will be denied in a future release
Traceback (most recent call last):
  File "/home/micha/Documents/01_work/git/tf_java_import_testing/tf-import/mnist/mnist_jumpy.py", line 17, in <module>
    tf_model = jp.TFModel(path + '/mnist.pb')
  File "/home/micha/Documents/01_work/git/tf_java_import_testing/tf-import/venv-deepmom/lib/python3.7/site-packages/jumpy/tf_model.py", line 22, in __init__
    self.sd = TFGraphMapper.getInstance().importGraph(filepath)
  File "jnius/jnius_export_class.pxi", line 906, in jnius.JavaMultipleMethod.__call__
  File "jnius/jnius_export_class.pxi", line 638, in jnius.JavaMethod.__call__
  File "jnius/jnius_export_class.pxi", line 715, in jnius.JavaMethod.call_method
  File "jnius/jnius_utils.pxi", line 93, in jnius.check_exception
jnius.JavaException: JVM exception occurred: class java.lang.String cannot be cast to class org.tensorflow.framework.GraphDef (java.lang.String is in module java.base of loader 'bootstrap'; org.tensorflow.framework.GraphDef is in unnamed module of loader 'app')

Насколько я понимаю, я получаю исключение jnius.JavaException, потому что классы java.lang.String и org.tensorflow.framework.GraphDef живут в разных контекстах, но я не знаю, как решить эту проблему (я совершенно незнаком с jnius ).

Любая помощь будет принята с благодарностью.

Это моя версия mnist_jumpy.py: 

import os
try:
     # from jnius import autoclass
     import jumpy as jp
except KeyError:
     os.environ['JDK_HOME'] = "/usr/lib/jvm/java-11-openjdk-amd64"
     os.environ['JAVA_HOME'] = "/usr/lib/jvm/java-11-openjdk-amd64"
     import jumpy as jp

from scipy import ndimage
# import numpy as np
import os

path = os.path.dirname(os.path.abspath(__file__))

# load tensorflow model
tf_model = jp.TFModel(path + '/mnist.pb')

# load jpg to numpy array
image = ndimage.imread(path + '/img (1).jpg').reshape((1, 28, 28))

# inference - uses nd4j
prediction = tf_model(image)  # prediction is a jumpy array

# get label from predction using argmax
label = jp.argmax(prediction.reshape((10,)))

print(label)
...