如何在 react-native 中加载自定义 tensorflowJS 模型？
How to load custom tensorflowJS model in react-native?
我已经使用 tensorflowjs_converter 将我的 Keras 模型转换为 TensorflowJS。它生成了一个 'model.json' 文件和 9 个 '*.bin' 文件。我正在尝试使用以下函数加载此模型。
const model = await tf.loadLayersModel(bundleResourceIO(modelJson, modelWeights));
modelWeights 只选择一个权重。但是我有9个。如何正确加载此模型？
好的，我刚刚找到了解决方案。基本上 tensorflow_converter 将权重分解成更小的尺寸。但是如果你增加权重文件的大小限制，生成的文件数量将会减少。假设您的模型权重文件需要 30MB。因此，将限制增加到 >=30MB，它将只生成一个权重文件。您可以按照以下命令执行此操作：
tensorflowjs_converter --input_format keras --weight_shard_size_bytes 60000000 'your_model.h5' 'output_dir'
我已经使用 tensorflowjs_converter 将我的 Keras 模型转换为 TensorflowJS。它生成了一个 'model.json' 文件和 9 个 '*.bin' 文件。我正在尝试使用以下函数加载此模型。
const model = await tf.loadLayersModel(bundleResourceIO(modelJson, modelWeights));
modelWeights 只选择一个权重。但是我有9个。如何正确加载此模型？
好的，我刚刚找到了解决方案。基本上 tensorflow_converter 将权重分解成更小的尺寸。但是如果你增加权重文件的大小限制，生成的文件数量将会减少。假设您的模型权重文件需要 30MB。因此，将限制增加到 >=30MB，它将只生成一个权重文件。您可以按照以下命令执行此操作：
tensorflowjs_converter --input_format keras --weight_shard_size_bytes 60000000 'your_model.h5' 'output_dir'