fs4now

Results 11 comments of fs4now

多谢,我这边因为机器配置低的原因,现训练模型所需时间太长了,想直接把ckpt的转化成pb的,因为我直接跑你的demo.py(摄像头调用的)成功了,对应的模型是savedmodel.ckpt,所以想把savedmodel.ckpt转换成pb的,在网上找了一下资料,有如下代码, import tensorflow as tf import os.path import argparse from tensorflow.python.framework import graph_util MODEL_DIR = "model/pb" MODEL_NAME = "frozen_model.pb" if not tf.gfile.Exists(MODEL_DIR): #创建目录 tf.gfile.MakeDirs(MODEL_DIR) def freeze_graph(model_folder): checkpoint = tf.train.get_checkpoint_state(model_folder) #检查目录下ckpt文件状态是否可用...

没关系的,麻烦你了,你说的第二条方法成功生成pb文件了,多谢 我测试一下看看效果再

因为我想把模型迁移到安卓上面,安卓端需要下面这三个变量名称,可以在train.py中添加进模型么? 模型中输入变量的名称(最开始的入口节点,是demo.py中第95行的input_image么?) 模型中输出变量的名称(最后的输出节点,没有找到) 输入值的大小(我看了一下demo.py中的95行,是[160, 160, 3]这个大小么?长*宽*三颜色通道数) 刚开始接触不到3天,望理解小白,非常感谢

好的,多谢指点,这两个名称能在train.py训练的时候就固化加入到pb模型中么?

嗯,好的,我再看看怎样才能迁移到安卓端吧,多谢了。

@shaonfu 你留个qq号吧 我加你

@315386775,我没有迁移,你们呢?

@bityangbing 这个我也不清楚,最后我采用的是谷歌官方迁移学习demo来实现区分男女的

好的,好的,多谢大佬指点