我想将rcmalli_vggface_tf_vgg16.h5预训练模型转换为js格式,以便在tensorflow-js项目中使用。
要转换此文件,我尝试了不同的方法,但我无法解决这个问题。
我使用converters.convert_tf_saved_model
方法加载,然后将其转换为json文件。
converters.convert_tf_saved_model('rcmalli_vggface_tf_vgg16.h5','web_model')
但每次显示以下错误:
SavedModel file does not exist at: rcmalli_vggface_tf_vgg16.h5
虽然我确信h5文件就在运行程序的文件旁边。
我尝试完整路径地址,但发生了相同的错误。我不知道是什么问题
1条答案
按热度按时间dwbf0jvd1#
您正在转换keras * saved_model *,但该模型是keras * layers_model *,因为它存储在h5文件中。
我发现这种方法转换模型最简单。而且它在大约2秒钟内就完成了。
1.转到TensorflowJS converter Github
1.遵循安装说明
1.输出将是一个层模型,必须使用
tf.loadLayers
API加载。1.例如,使用
const model = await tf.loadGraphModel('path/to/model.json');
注 *:
./converted
是输出目录,请确保不要覆盖自己的东西。