如何将自定义数据集导入TensorFlow

时间:2019-10-14 15:56:41

标签: tensorflow dataset

对于我在uni的第一次工作,我们获得了一个充满交通标志的数据集,该数据集被组织成43个不同标志的文件夹。我不知道如何将给定的数据集导入张量流,文件是.ppm

1 个答案:

答案 0 :(得分:0)

将数据放入NumPy数组中,然后可以轻松地与Tensorflow Keras一起使用。 tf.keras.Model.fit()tf.keas.Model.Predict()可以采用Numpy数组。

.ppm格式似乎是一种非常简单的文本格式,每行一个像素值。因此,将每一行读入列表或直接读入NumPy数组。阅读时可能需要跳过标题行。看到这个问题How to read a file line-by-line into a list?