Hello everyone,
I’ve been trying to load a model that accept 2 images as input source and passed as a list. I got this error:
Process: org.pytorch.demo.imagesegmentation, PID: 10778
java.lang.ArrayIndexOutOfBoundsException: vector
at org.pytorch.LiteNativePeer.forward(Native Method)
at org.pytorch.Module.forward(Module.java:52)
at org.pytorch.imagesegmentation.MainActivity.run(MainActivity.java:140)
Code to load model:
I tried this one;
Map<String, IValue> outTensors = mModule.forward(
IValue.tupleFrom(
IValue.from(inputTensor),
IValue.from(inputTensor2)
)
).toDictStringKey();
and this
IValue img = IValue.listFrom(inputTensor, inputTensor2);
final Tensor outputTensor = mModule.forward(img).toTensor();
Please, someone could help me?