由于这个问题,我无法入睡。
操作系统:window10 tensorflow版本:1.13.1
源代码生成android应用程序
'''
from tensorflow.python.tools import freeze_graph
MODEL_NAME = 'har'
input_graph_path = 'checkpoint/' + MODEL_NAME+'.pbtxt'
checkpoint_path = './checkpoint/' +MODEL_NAME+'.ckpt'
restore_op_name = "save/restore_all"
filename_tensor_name = "save/Const:0"
output_frozen_graph_name = 'frozen_'+MODEL_NAME+'.pb'
freeze_graph.freeze_graph(input_graph_path, input_saver="",
input_binary=False, input_checkpoint=checkpoint_path,
output_node_names="y_", restore_op_name="save/restore_all",
filename_tensor_name="save/Const:0",
output_graph=output_frozen_graph_name, clear_devices=True, initializer_nodes="")
'''
和android stdio代码
'''
package io.github.introml.activityrecognition;
import android.content.Context;
import org.tensorflow.contrib.android.TensorFlowInferenceInterface;
public class TensorFlowClassifier {
static {
System.loadLibrary("tensorflow_inference");
}
private TensorFlowInferenceInterface inferenceInterface;
private static final String MODEL_FILE = "file:///android_asset/frozen_har.pb";
// private static final String MODEL_FILE = "file:///android_asset/frozen_har.pb";
private static final String INPUT_NODE = "inputs";
private static final String[] OUTPUT_NODES = {"y_"};
private static final String OUTPUT_NODE = "y_";
private static final long[] INPUT_SIZE = {1, 200, 3};
private static final int OUTPUT_SIZE = 6;
public TensorFlowClassifier(final Context context) {
inferenceInterface = new TensorFlowInferenceInterface(context.getAssets(), MODEL_FILE);
}
public float[] predictProbabilities(float[] data) {
float[] result = new float[OUTPUT_SIZE];
inferenceInterface.feed(INPUT_NODE, data, INPUT_SIZE);
inferenceInterface.run(OUTPUT_NODES);
inferenceInterface.fetch(OUTPUT_NODE, result);
//Downstairs Jogging Sitting Standing Upstairs Walking
return result;
}
}
'''
我听不懂
应用程序立即终止
应用程序立即终止