【发布时间】:2023-03-28 15:16:01
【问题描述】:
** 我知道类似的问题!! **
我的问题是针对我的特殊情况... 我使用 Google Vision 训练我自己的模型来检测自定义对象。 过去我遇到过关于形状的类似错误,我通过重塑输入图像解决了这些错误。
这个特殊错误告诉我我的形状必须是空数组或空形状。这甚至可能吗?如果这不是故障,我该如何解决?
这就是我在其他项目中抱怨形状时解决以前错误的方法。此解决方案不适用于空数组/形状
const model = await autoML.loadObjectDetection('./model/model.json');
// const model = await tfjs.loadGraphModel('./model/model.json');
await tfjs.ready();
const tfImg = tfjs.browser.fromPixels(videoElement.current).expandDims(0);
const smallImg = await tfjs.image.resizeBilinear(tfImg, [224, 224]);
const resized = tfjs.cast(smallImg, 'float32');
const t4d = tfjs.tensor4d(Array.from(resized.dataSync()), [1, 224, 224, 3]);
const predictions = await modelRef.current.detect(tfImg, options);
【问题讨论】:
-
转换模型时一定有问题,因为即使在model.json中输入形状是“tensorShape”:{“dim”:[{“size”:“-1”},{ "size": "-1"}, {"size": "-1"}, {"size": "3"}]}}} 这没有意义
-
感谢您澄清这一点。作为新手,我不确定我的假设是否正确。此外,我注意到 dict.txt 中所有标签旁边的符号或字母“d”。当 3 小时的训练结束并且 google vision 决定结束对我的模型的训练时,似乎正在处理某些事情。两次
-
我建议在 tensorflow.js 或 google vision 的 github 页面上打开这个 issue。
-
@DT 还没有...我会联系谷歌的 automl 视觉团队,看看他们是否有解决方案。
标签: tensorflow tensorflow2.0 tensorflow.js google-vision automl