批量大小为 0 或 NaN。请选择一个非零分数

Question

在使用 tensorflow.js 时，我们正在添加图像进行训练，但在训练这些图像时出现错误。

        const gorra = new Image();
        gorra.src = "https://ml5js.org/docs/assets/img/bird.jpg";
        gorra.width = 224;
        gorra.height = 224;
        console.log("adding images DONE", gorra);

        img2 = new Image();
        img2.src = "{!$Resource.cat}"
        img2.width = 224;
        img2.height = 224;
        console.log(img2);

        console.log("setup classifier");

        console.log("adding example image...");
        const ex = await classifier.addImage(document.getElementById('imgshow'), "Gorra");
        console.log("adding ex image DONE!...", ex);
        const ex1 = await classifier.addImage(img2, "Gorra1");
        console.log("adding ex1 image DONE!...", ex1);

        console.log("Training");
        const trainer = await classifier.train();
        console.log("Training DONE", trainer);

请帮我找出这些图片添加到哪里以及为什么会出现错误。

塔克

Answer 1

我认为您的第二个图像源不正确，或者该图像未加载到内存中，并且您正在对一张图像进行训练，并且分类器需要至少 2 张图像用于训练目的。 另请参阅： https://github.com/ml5js/ml5-library/issues/212