tensorflow模型压缩——ckpt转pb

本文讲述了将TensorFlow的ckpt模型转换为pb格式的过程,探讨了在转换过程中遇到的困难,特别是对于回归模型输出定义的问题。作者指出,pb格式虽简化了文件管理,但模型大小变化不大,可能需要进一步转换为tf lite。文中还提到,由于模型的运算并非全部由tf定义,因此寻找正确的输入输出定义成为挑战,并分享了在定义网络结构时可能需要重新考虑的方面。
摘要由CSDN通过智能技术生成

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

  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 3
    评论
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

小李飞刀李寻欢

您的欣赏将是我奋斗路上的动力!

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值