我们可以将keras模型(.h5-多个组)文件转换成一个二进制文件-keras-js

时间:2019-10-19 15:11:36

标签: node.js keras binary hdf5

我生成了我的model.h5和model.json文件。我想将其与节点js代码一起使用。 npm keras-js。我需要一个包含所有数据集的二进制文件(.bin)

const KerasJS = require('keras-js')
const model = new KerasJS.Model({
    filepath: __dirname + '/model/dset.model.bin',
    headers: {},
    filesystem: true,
    gpu: false,
    transferLayerOutputs: false,
    pauseAfterLayerCalls: true,
    visualizations: []
})
h5dump 
-a /layer_names 
-a /dense_1/weight_names 
-d /dense_1/dense_1/bias:0 
-d /dense_1/dense_1/kernel:0 
-a /dense_2/weight_names  
-d /dense_2/dense_2/bias:0 
-d /dense_2/dense_2/kernel:0 
-b LE -o dset.model.bin model.h5

output -> dset.model.bin gives error
Error: invalid wire type 4 at offset 1

0 个答案:

没有答案