react-native - 如何在 react-native 中加载自定义 tensorflowJS 模型？

Question

我已经使用 tensorflowjs_converter 将我的 Keras 模型转换为 TensorflowJS。它生成了一个“model.json”文件和 9 个“*.bin”文件。我正在尝试使用以下功能加载此模型。

const model = await tf.loadLayersModel(bundleResourceIO(modelJson, modelWeights));

modelWeights 只选择一个权重。但是我有9个。如何正确加载此模型？

Answer 1

好的，我刚刚找到了解决方案。基本上 tensorflow_converter 将权重分解为一些更小的尺寸。但是如果增加权重文件的大小限制，生成的文件数量会减少。假设您的模型权重文件占用 30MB。因此，将限制增加到 >=30MB，它只会生成一个权重文件。您可以按照以下命令执行此操作:

tensorflowjs_converter --input_format keras --weight_shard_size_bytes 60000000 'your_model.h5' 'output_dir'