tensorflow相关

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数组参数代表维数

文章目录
  1. 1. Android 模型移植
|