python - 在 Tensorflow 中使用您自己的数据

Question

我已经知道如何使用 mnist 数据集制作神经网络。 3 个月来，我一直在寻找有关如何在您自己的数据集上训练神经网络的教程，但我就是不明白。如果有人可以建议任何好的教程或解释所有这些是如何工作的，请提供帮助。附言。我不会安装 NLTK。似乎很多人都在文本上训练他们的神经网络，但我不会那样做。如果我要安装 NLTK，我只会使用它一次。

Answer 1

我建议您使用 OpenCV 库。无论您使用 MNIST 数据还是 PIL，当它被加载时，它们都只是 NumPy 数组。如果你想让 MNIST 数据集适合你训练的模型，我是这样做的:

1.使用cv2.imread加载所有你希望它们作为训练数据集的图像。

2.使用cv2.cvtColor将所有图片转为灰度图，并resize为28x28。

3.将所有数据集中的每个像素除以255。

4.照常训练!

我没试过让它成为你自己的格式，但理论上它是一样的。