tensorflow - IllegalStateException: エラー: セッションを実行できません arg[0] は uint8 であると予想されますが、int32 が提供されています || サポートされていないデータ型: UBYTE

Question

Tensorflow 1.13.1、org.deeplearning4j "1.0.0-beta6" (Tensorflow 1.15) を使用してトレーニング済みモデル Faster-RCNN で RunGraphExample frozen_graph.pb を実行しようとしています

val data: Array[Array[Int]] = new Array[Array[Int]](img.getWidth * img.getHeight);
    for (i <- 0 until img.getWidth) {
      for (j <- 0 until img.getHeight()) {
        val ar: Array[Int] = new Array(3)
        ar(0) = color.getRed.byteValue() & 0xff
        ar(1) = color.getGreen.byteValue() & 0xff
        ar(2) = color.getBlue.byteValue() & 0xff
        data(i * img.getHeight + j) = ar
      }
    }
var arr: INDArray = Nd4j.createFromArray(data) 
//.castTo( org.nd4j.linalg.api.buffer.DataType.UBYTE)

inputMap.put(inputs.get(0), ShapedArray)行にエラーが表示されます

shapeArray が int の場合、エラーは次のとおりです。

Unable to run session Expects arg[0] to be uint8 but int32 is provided

shapeArray が short または .castTo(UBYTE) の場合、エラーは次のとおりです。

Unsupported data type: UBYTE
or
Unsupported data type: SHORT

約 1.0.0-beta3 から、Deeplearning4j はval コンプレッサーのサポートを削除しました = BasicNDArrayCompressor.getInstance.setDefaultCompression("UINT8")

uint8 DataType を使用してイメージを INDArray に正しく変換するにはどうすればよいですか?

https://github.com/eclipse/deeplearning4j-examples/blob/master/tf-import-examples/src/main/java/org/nd4j/examples/RunGraphExample.java