ckpt模型转pb是项目落地的关键,如果你仅仅是想在服务器部署,也可忽略此步。人生的艰难有很多坑,不从坑里爬出来体会不到生命中的乐趣所在。简单的说,原来ckpt模型有4个文件(分别为checkpoint,index,data,meta),转成pb后就一个文件,至少方便了很多。然而其大小并无明显变化,甚至没有变化,想要有明显变化,还需要转tf lite,但首先第一步是转成pb格式。
pb格式的转化其实也有很多参考,然后不同模型的inference写起来可能不是那么简单,在其中都是要定义一个入口和一个出口,也就是必需input和output,这样图结构才可以方便的拿起来用。本人先前所见的都是分类模型的pb转化,未见有回归模型的pb转化,有人说这不都一样吗?你别瞎逼逼,你先甩给我个回归的链接,我跪谢你。
其中的难点在于回归的inference中output不好定义,而分类的只是定义一个softmax就OK了。目前只有发挥我的聪明才智,或者广泛搜索了。
人生不易,请你珍惜眼前的我。随时更新,欢迎关注。
补充一:寡人得到了模型的节点名字,但是也不知道该怎么定义输出,因为模型中的输出结果并不是想要的直接输出,还要经过运算,而这种运算不是tf定义的,是不是先定义下tf的加减乘除为好?
tensor_name: beta1_power
tensor_name: beta2_power
tensor_name: net/rnn/multi_rnn_cell/cell_0/gru_cell/candidate/bias/Adam
tensor_name: net/rnn/multi_rnn_cell/cell_0/gr