Question

我使用以下命令从https://github.com/tensorflow/hub/blob/master/examples/image_retraining/retrain.py使用retrain.py创建了一个自定义tensorflow lite模型

python retrain.py --image_dir newImageDirectory --tfhub_module https://tfhub.dev/google/imagenet/mobilenet_v2_100_224/feature_vector/1

然后我使用toco将output_graph.pb文件转换为lite文件。使用以下命令

bazel run tensorflow/contrib/lite/toco:toco -- --input_file=/tmp/output_graph.pb --output_file=/tmp/optimized.lite --input_format=TENSORFLOW_GRAPHDEF --output_format=TFLITE --inpute_shape=1,224,224,3 --input_array=input --output_array=final_result --inference_type=FLOAT --input_data_type=FLOAT

然后我获取新的lite文件和labels.txt文件，并将它们放入poets 2 https://github.com/googlecodelabs/tensorflow-for-poets-2的tensorflow中，看看我是否可以让它开始对新类别进行分类。应用程序启动时，我收到以下错误。

Caused by: java.lang.NullPointerException: Can not allocate memory for the interpreter                                                                                            at org.tensorflow.lite.NativeInterpreterWrapper.createInterpreter(Native Method)
                                                                      at org.tensorflow.lite.NativeInterpreterWrapper.<init>(NativeInterpreterWrapper.java:63)
                                                                                        at org.tensorflow.lite.NativeInterpreterWrapper.<init>(NativeInterpreterWrapper.java:51)
                                                                                        at org.tensorflow.lite.Interpreter.<init>(Interpreter.java:90)
                                                                                        at com.example.android.tflitecamerademo.ImageClassifier.<init>(ImageClassifier.java:97)

Answer 1

这是由于您使用的计算机造成的，请确保您有足够的内存来分配给程序，或者尝试增加交换文件的大小？

Tensorflow自定义TFLite java.lang.NullPointerException：无法为解释器分配内存

1 个答案: