如何将.hdf5转换为.h5 keras模型

时间:2020-08-29 08:38:26

标签: tensorflow keras

我有一个here中使用的经过预先训练的hdf5模型背景去除模型。我希望将其转换为h5,因为coremltools转换器需要该类型。

到目前为止,coremltools python脚本给出以下错误:

NotImplementedError: Expected model format: [SavedModel | [concrete_function] | tf.keras.Model | .h5], got main_model.hdf5

1 个答案:

答案 0 :(得分:2)

实际上是相同的格式,因此您可以通过将文件重命名为.h5来更改扩展名,并且应该可以正常工作。