2

.txt ファイルから多くの「画像」を読み込む必要があり、それらを使用してテンソルフロー データセットを生成したいと考えています。現在、numpy.loadtxt を使用してすべての単一のマトリックスを読み取り、形状 [N_matrices、高さ、幅、N_channels] の配列と、すべてのマトリックスのラベルを持つ同様の配列を作成します。

以下を使用して、これら 2 つの配列からテンソルフロー データセットを作成します。

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})

次の関数を使用して、このデータセットからバッチを作成したいと思います (ここで行ったように):

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)

ファイル "/home/.local/lib/python3.5/site-packages/tensorflow/contrib/slim/python/slim/data/dataset_data_provider.py"、85 行目、init dataset.data_sources 内、

AttributeError: 'TensorSliceDataset' オブジェクトに属性 'data_sources' がありません

なぜこのエラーが発生するのですか?どうすれば修正できますか? また、txt ファイルから tensorflow (または tensorflow-slim) への入力を処理するためのより良い方法があると思いますが、これに関する情報はほとんど見つかりませんでした。より良い方法でデータセットを生成するにはどうすればよいですか?

4

0 に答える 0