keras模型转pb格式

论坛 期权论坛 编程之家     
选择匿名的用户   2021-6-1 11:53   425   0

千万不能用tf.keras保留的模型h5文件去转pb文件,各种坑!

但是可以用tf.keras训练模型,保存权重文件之后,再用keras加载网络文件和权重,再保留成模型h5文件,注意的是网络文件里面的定义也要换成keras的层。

之后keras转pb可直接用https://github.com/amir-abdi/keras_to_tensorflow里面的keras_to_tensorflow.py转换。

Unknown: Failed to get convolution algorithm的问题参考https://blog.csdn.net/weixin_43820996/article/details/102906434

分享到 :
0 人收藏
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

积分:3875789
帖子:775174
精华:0
期权论坛 期权论坛
发布
内容

下载期权论坛手机APP