如何将图像加载到 tensorflow.js 代码中,该代码读取手写数字并对其进行分类

人工智能 机器学习 卷积神经网络 图像识别
2021-11-01 23:13:15

我是机器学习的新手,所以我想我应该看看谷歌的张量流指南,我知道如何在 JS 中编码,所以这就是我使用 tensorflow.js 的原因,指南中有一些例子可以训练它识别手写来自 MNIST 手写数据集的数字,我有点理解代码中发生了什么,但由于我对 ML 很陌生,所以它不是很多,我浏览了代码,发现它没有逐个图像地训练自己但它要求一个包含所有图像的精灵,然后将其切割成它需要的东西,从性能的角度来看这是有道理的,但由于这个过程有点抽象,我不明白到底发生了什么,我想上传我自己的图像并调用模型的预测器,但我不知道该怎么做,有什么帮助吗?

我在想,在 28x28 的画布上绘制一个数字可能也很有趣,而不是上传图像,但我需要知道如何在使用我自己的数据训练模型后对其进行测试。

教程:https ://js.tensorflow.org/tutorials/mnist.html

1个回答

要测试模型,您必须获得与训练数据具有相同维度的张量。对于同一张图片,这个张量应该是相同的,无论你用同一张图片重新制作张量多少次。话虽如此,您可以尝试对张量的制作方式进行逆向工程。提示:输入张量的维度应该是 (28, 28, 1)。