Hello community,
I am new in TensorFlow. I am trying to execute this open-source code. However, I am getting this error while preparing the dataset;
def prepare_dataset(images, labels):
images_placeholder = tf.compat.v1.placeholder(tf.float32, images.shape)
labels_placeholder = tf.compat.v1.placeholder(tf.int64, labels.shape)
dataset = tf.data.Dataset.from_tensor_slices((images_placeholder, labels_placeholder))
dataset = dataset.shuffle(buffer_size=10000, seed=config['random_seed']).repeat()
if config['augment_dataset']:
dataset = dataset.map(
make_data_augmentation_fn(
standardization=config['augment_standardization'],
flip=config['augment_flip'],
padding=config['augment_padding'],
is_training=True))
dataset = dataset.batch(batch_size=50)
print(dataset,"this is batch")
# iterator = tf.compat.v1.dataset.make_initializable_iterator()
iterator = tf.compat.v1.data.make_initializable_iterator(dataset)
return (images_placeholder, labels_placeholder), dataset, iterator
clean_placeholder, clean_train_dataset_batched, clean_training_iterator = prepare_dataset(clean_train_images, clean_train_labels)
print(clean_train_dataset_batched)
poisoned_placeholder, _, poisoned_training_iterator = prepare_dataset(poisoned_train_images, poisoned_train_labels)
if len(poisoned_train_indices) > 0:
poisoned_only_placeholder, _, poisoned_only_training_iterator = prepare_dataset(poisoned_only_train_images, poisoned_only_train_labels)
poisoned_no_trigger_placeholder, _, poisoned_no_trigger_training_iterator = prepare_dataset(poisoned_no_trigger_train_images, poisoned_no_trigger_train_labels)
handle = tf.compat.v1.placeholder(tf.string, shape=[])
# input_iterator = tf.compat.v1.data.Iterator.from_string_handle(handle,clean_train_dataset_batched.output_types,
# clean_train_dataset_batched.output_shapes)
The error is batches are not append
<BatchDataset shapes: ((None, 32, 32, 3), (None,)), types: (tf.float32, tf.int64)> this is batch
<BatchDataset shapes: ((None, 32, 32, 3), (None,)), types: (tf.float32, tf.int64)>
<BatchDataset shapes: ((None, 32, 32, 3), (None,)), types: (tf.float32, tf.int64)> this is batch
Can anyone help me in solving this error? Appreciate your time and effort