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

时间:2019-01-22 07:03:04

标签: javascript tensorflow artificial-intelligence

虽然使用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);

请帮助我找到添加这些图像的位置以及为什么会出现该错误。

Thak

1 个答案:

答案 0 :(得分:0)

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