Как по-разному реплицировать каждый элемент с помощью API набора данных tenorflow?

Question

Я создаю набор данных тензорного потока train_image, который имеет 10 изображений класса A и 20 изображений класса B. Теперь, как я могу реплицировать каждый класс отдельно?

Я хочу, чтобы train_image включал 100 изображений класса A (10 повторов на каждом изображении класса A) и 100 изображений класса B (5 повторов на каждом изображении класса B),

Answer 1

Используйте увеличение данных ImageDataGenerator, оно преобразует изображения. Набор преобразований включает в себя вращение, масштабирование и т. Д., Которые являются каким-то образом новыми данными. Это зависит от того, сколько эпох вы запустили, будет создано столько изображений в зависимости от steps_per_epoch. Эти дополненные изображения не сохраняются в памяти, они генерируются во время работы и теряются после тренировки.

datagen = keras.preprocessing.image.ImageDataGenerator(
featurewise_center=True,
featurewise_std_normalization=True,
rotation_range=20,
width_shift_range=0.2,
height_shift_range=0.2,
horizontal_flip=True)