Open sachin-ranka opened 5 years ago
Maybe your model is a float model.
Try to set the QUANT as FALSE and give a try & Instead of using byte[][] use the Float[][] in your Classifier class. I had a similar issue and its fixed now. i.e; private static final boolean QUANT = false; & float[][] result = new float[1][labelList.size()]; interpreter.run(byteBuffer, result); return getSortedResultFloat(result);
The input type is INT32 so its multiplied with 4. byteBuffer = ByteBuffer.allocateDirect(4 BATCH_SIZE inputSize inputSize PIXEL_SIZE);
Replaced default graph.lite with my custom trained model, followed steps as mentioned in Codelabs
Error log:
Unable to run the model in android.