Пример кода Java прерывается при увеличении младшей версии Tensorflow ⇐ JAVA

[Anonymous]

Я создал этот код на основе примеров Java:
https://github.com/tensorflow/java-mode ... mples/src/ main/java/org/tensorflow/model/examples/cnn/fastrcnn/FasterRcnnInception.java
У меня все работает. И затем я заметил, что используемая версия Tensorflow довольно старая: 0.4.0:
https://github.com/tensorflow/java-mode ... om.xml#L15
Поэтому увеличил его до 0.4.1, думая, что «выпуск патча не должен изменить ничего важного».
И:

Код: Выделить всё

Exception in thread "main" org.tensorflow.exceptions.TFInvalidArgumentException: image must be 3-dimensional[1,900,1352,3]
[[{{node EncodeJpeg}}]]
at org.tensorflow.internal.c_api.AbstractTF_Status.throwExceptionIfNotOK(AbstractTF_Status.java:87)
at org.tensorflow.Session.run(Session.java:850)
at org.tensorflow.Session.access$300(Session.java:82)
at org.tensorflow.Session$Runner.runHelper(Session.java:552)
at org.tensorflow.Session$Runner.runNoInit(Session.java:499)
at org.tensorflow.Session$Runner.run(Session.java:495)
at HelloTensorFlow.main(FasterRcnnInception.java:259)

Я пытался понять, что делает encodeJpeg (отсюда:
https://github.com/tensorflow/java-mode ... ensorflow- example/src/main/java/org/tensorflow/model/examples/cnn/fastrcnn/FasterRcnnInception.java#L319 )

Код: Выделить всё

        tf.image.encodeJpeg(
tf.dtypes.cast(tf.reshape(
tf.math.mul(
tf.image.drawBoundingBoxes(tf.math.div(
tf.dtypes.cast(tf.constant(reshapeTensor),
TFloat32.class),
tf.constant(255.0f)
),
boxesPlaceHolder, colors),
tf.constant(255.0f)
),
tf.array(
imageShape.asArray()[0],
imageShape.asArray()[1],
imageShape.asArray()[2]
)
), TUint8.class),
jpgOptions));

Но я слишком новичок, чтобы понимать, что происходит. Кто-нибудь может объяснить? И, возможно, объясните, что изменилось между 0.4.0 и 0.4.1, чтобы сломать это.

Подробнее здесь: https://stackoverflow.com/questions/788 ... -increment