【发布时间】:2021-07-30 14:52:38
【问题描述】:
我正在使用可教机器导出模型。
const modelJson = require("./assets/models/model.json");
const modelWeights = require("./assets/models/weights.bin");
useEffect(() => {
let checkTf = async () => {
await tf.ready();
const model = await tf.loadLayersModel(bundleResourceIO(modelJson, modelWeights));
setModel(model);
setDisplayText("loaded Models");
setTfReady(true);
};
checkTf();
}, []);
在预测图像异步函数中加载的模型:
const imageTensor = images.next().value;
const prediction = (await model.predict())[0]; //ERROR
const highestPropPred = prediction[0];
无法加载预测此错误:
[Unhandled promise rejection: Error: Error when checking : expected input_1 to have 4 dimension(s), but got array with shape [200,152,3]]
【问题讨论】:
标签: react-native machine-learning deep-learning