转换rcmalli_vggface_tf_vgg16.h5至TF.js时遇SavedModel文件不存在错误
解决H5模型转TensorFlow.js格式的报错问题
你遇到的报错核心原因是用错了转换方法:converters.convert_tf_saved_model专门用于转换SavedModel格式的模型,而你手里的rcmalli_vggface_tf_vgg16.h5是Keras的HDF5格式模型,所以该方法会认为找不到符合要求的SavedModel文件。
正确的转换方式有两种:
方式一:命令行转换(推荐)
确保已经安装tensorflowjs包:
pip install tensorflowjs
然后执行转换命令:
tensorflowjs_converter --input_format keras rcmalli_vggface_tf_vgg16.h5 web_model
方式二:Python代码转换
改用针对Keras模型的转换方法convert_keras_model,代码如下:
import tensorflowjs as tfjs tfjs.converters.convert_keras_model('rcmalli_vggface_tf_vgg16.h5', 'web_model')
执行上述任一操作后,web_model目录下会生成model.json和对应的权重文件,这些就是TensorFlow.js项目可以直接加载的模型文件。
内容的提问来源于stack exchange,提问作者Ahmad Badpey
相关产品推荐
相关产品推荐

