.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) への入力を処理するためのより良い方法があると思いますが、これに関する情報はほとんど見つかりませんでした。より良い方法でデータセットを生成するにはどうすればよいですか?