Android 模型移植
将模型移植到手机端的时候,需要提供input和output的layer名字
tensorflow保存为pb模型的时候output名字在layer层里面会加入后缀,我们需要查看后缀之后输入它的输出名字
(如Dense层,名字为output的层的输出名字是output/Sigmoid
可以通过下述代码查看层的名字,这在将层移植到手机端上设置输入输出名的时候会很有用[op for op in sess.graph.get_operations() if op.type == "Const"]
也可以在tensorflow里先生成图然后在tensorboard里查看,里面的名字也是一样的(相对更加方便看一些)
Android版本的tensorflow的feed输入是一维float数组,最后的long数组参数代表维数