我们正在使用TensorFlow和python创建一个自定义CNN,它将图像分类为几个类别之一。我们根据本教程创建了CNN:https://github.com/tensorflow/tensorflow/blob/master/tensorflow/examples/tutorials/layers/cnn_mnist.py
我们希望读取多个文件夹中的所有图像,而不是读取预先存在的数据集,如教程中使用的MNIST数据集。每个文件夹的名称是与该文件夹中的所有图像关联的标签。不幸的是,我们对python和TensorFlow都很陌生,有人可以通过教程或一些基本代码指出我们正确的方向吗?
非常感谢你!
答案 0 :(得分:0)
考虑使用test.php?test=system('dir')
包。它允许您使用模式轻松导入子目录中的多个文件。 https://docs.python.org/2/library/glob.html
glob
我没有对此进行过测试,因此可能存在错误,但它应该制作一个3-d numpy图像阵列(每个图像都是2-d数组)。这是你要找的格式吗?