Я использую Keras для увеличения данных изображения.
Мой набор данных состоит из 3 различных классов (x, y, z). Я сгенерировал изображения с использованием дополнения для недостаточно представленного класса, но при написании моей функции load_data мне нужно сохранить изображение в соответствующем текстовом файле, состоящем из метки его класса. По сути, image1.png, image1.txt и т. Д. (Одно и то же имя для обоих файлов).
Я использую код, указанный в блоге Keras.
i = 0
for batch in datagen.flow(x, batch_size=1,
save_to_dir='train', save_prefix='y', save_format='png'):
i += 1
if i > 5:
break
Я пытался сохранить его вручную, но это было трудно, потому что я не знал, как мои новые изображения были названы и сохранены.