转换自,H5型号为,pt model：将模型从keras h5转换为pytorch

at0kjp5o 于 2023-04-30 发布在其他

关注(0)|答案(1)|浏览(330)

如何使用APytorch文件中的h5模型;我使用Keras和TensorFlow后端工作，所以这里是我保存的模型：

model = tf.keras.applications.ResNet50(include_top=False, weights=None, input_tensor=tf.keras.Input(shape=(224, 224, 3)), pooling=None)
....#training
model.save("mymodel.h5") 
model.save_weights("saved_weights.h5")

现在我迁移到pytorch，我想使用保存的模型。那么我如何转换它。谢谢

pytorch

来源：https://stackoverflow.com/questions/76091366/converting-from-h5-model-to-pt-model-convert-model-from-keras-h5-to-pytorch

1条答案

按热度按时间

izkcnapc1#

一种方法是将tensorflow模型转换为ONNX框架。然后将ONNX模型转换为Torch。
安装tensorflow到onnx转换器

pip install onnxruntime
pip install git+https://github.com/onnx/tensorflow-onnx

将tonsorflowsaved_model转换为onnx：

python -m tf2onnx.convert --saved-model model_tensorflow --output model.onnx --opset 11 --verbose

将ONNX转换为Pytorch：
尝试使用this repository。

赞(0）回复(0）举报 2023-04-30

我来回答

转换自,H5型号为,pt model：将模型从keras h5转换为pytorch

1条答案

相关问题

热门标签

最新问答