Я сгенерировал дата сет из директории
training_dateset = image_dataset_from_directory(
'Fruit-Images-Dataset-master/Training',
subset='training',
seed=42,
validation_split=0.1,
batch_size=batch_size,
image_size=image_size
)
Я хочу обработать каждый элемент изображения что бы он лучше читался нейросетью
SIZE = 224
def resize_image(img, label):
img = img.numpy().astype("uint8")
img = tf.cast(img, tf.float32)
img = tf.image.resize(img, (SIZE, SIZE)) # изменить размер
img = img / 255.0 # Переумножить значение что бы оно было равно 0.255
return img, label
Но когда я применяю
train_resized = training_dateset[0].map(resize_image) # Изменить размер каждой картинки
То получаю
> 'PrefetchDataset' object is not subscriptable?
В чем проблема и как это решить?