千万不能用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 |