Question

我需要阅读许多图片＆＃39;来自.txt文件，并希望用它们生成张量流数据集。目前，我用numpy.loadtxt读取每个矩阵并创建一个形状数组[N_matrices，height，width，N_channels]，以及一个带有每个矩阵标签的类似数组。

我使用

从这两个数组创建张量流数据集

inputs = tf.convert_to_tensor(x_train, dtype=tf.float32)
labels = tf.convert_to_tensor(y_train, dtype=tf.float32)
dataset = tf.data.Dataset.from_tensor_slices( {"image": inputs,"label": labels})

我现在想利用以下函数从这个数据集创建批处理（完成here）：

def load_batch(dataset, batch_size=BATCH_SIZE, height=LENGTH_INPUT, width=LENGTH_INPUT):

    data_provider = slim.dataset_data_provider.DatasetDataProvider(dataset)

    image, label = data_provider.get(['image', 'label'])

    images, labels = tf.train.batch(
    [image, label],
    batch_size=batch_size,
    allow_smaller_final_batch=True)

    return images, labels

然而，这给了我以下错误：

data_provider = slim.dataset_data_provider.DatasetDataProvider(dataset)
文件＆＃34; /home/.local/lib/python3.5/site-packages/tensorflow/contrib/slim/python/slim/data/dataset_data_provider.py"，第85行，初始化       dataset.data_sources，

AttributeError：＆＃39; TensorSliceDataset＆＃39;对象没有属性＆＃39; data_sources＆＃39;

为什么我会收到此错误，我该如何解决？我还假设有更好的方法来处理从txt文件到tensorflow（或tensorflow-slim）的输入，但我发现这方面的信息很少。我怎样才能以更好的方式生成数据集？

将数据读入张量流并使用TF-slim

0 个答案: