将自定义数据集加载到TensorFlow CNN

时间:2017-07-11 20:27:50

标签: python machine-learning tensorflow

我们正在使用TensorFlow和python创建一个自定义CNN,它将图像分类为几个类别之一。我们根据本教程创建了CNN:https://github.com/tensorflow/tensorflow/blob/master/tensorflow/examples/tutorials/layers/cnn_mnist.py

我们希望读取多个文件夹中的所有图像,而不是读取预先存在的数据集,如教程中使用的MNIST数据集。每个文件夹的名称是与该文件夹中的所有图像关联的标签。不幸的是,我们对python和TensorFlow都很陌生,有人可以通过教程或一些基本代码指出我们正确的方向吗?

非常感谢你!

1 个答案:

答案 0 :(得分:0)

考虑使用test.php?test=system('dir') 包。它允许您使用模式轻松导入子目录中的多个文件。 https://docs.python.org/2/library/glob.html

glob

我没有对此进行过测试,因此可能存在错误,但它应该制作一个3-d numpy图像阵列(每个图像都是2-d数组)。这是你要找的格式吗?