在Tensorflow中使用您自己的数据

时间:2017-04-02 14:54:21

标签: python tensorflow neural-network dataset

我已经知道如何使用mnist数据集创建神经网络。我一直在寻找关于如何在你自己的数据集上训练神经网络3个月的教程,但我只是没有得到它。如果有人可以建议任何好的教程或解释所有这些是如何工作的,请帮助。 PS。我不会安装NLTK。似乎很多人都在训练他们的神经网络,但我不会这样做。如果我要安装NLTK,我只会使用一次。

1 个答案:

答案 0 :(得分:2)

我建议你使用OpenCV库。无论你使用什么MNIST数据或PIL,当它加载时,它们都只是NumPy数组。如果你想让MNIST数据集与训练有素的模型相匹配,我就是这样做的:

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

2.使用cv2.cvtColor将所有图像转换为灰度图像,并将其调整为28x28。

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

4.像往常一样进行训练!

我没有尝试过制作你自己的格式,但理论上它是一样的。