Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- # Сделайте предобработку данных: поделите значения яркости на 255. Для этого передайте специальный аргумент в загрузчик ImageDataGenerator (найдите в документации), в котором будет указано, что делить нужно на 255.
- # Создавать и обучать модель не нужно. На экране напечатайте значения пикселей изображения из первого батча обучающей выборки.
- from tensorflow.keras.preprocessing.image import ImageDataGenerator
- datagen = ImageDataGenerator(validation_split=0.25, rescale=1/255)
- train_datagen_flow = datagen.flow_from_directory(
- '/datasets/fruits_small/',
- target_size=(150, 150),
- batch_size=16,
- class_mode='sparse',
- subset='training',
- seed=12345)
- val_datagen_flow = datagen.flow_from_directory(
- '/datasets/fruits_small/',
- target_size=(150, 150),
- batch_size=16,
- class_mode='sparse',
- subset='validation',
- seed=12345)
- features, target = train_datagen_flow.next()
- print(features[0])
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement