Увеличение в Керасе - Генерация изображений - PullRequest
Новая
       59

Увеличение в Керасе - Генерация изображений

0 голосов
/ 07 января 2020

У меня несбалансированный набор данных, поэтому я решил использовать генератор изображений Keras.

Я создал фрагмент кода, чтобы три изображения были объединены в одно изображение, 

data_gen = ImageDataGenerator(rotation_range=10, 
                          brightness_range=(0.2, 1.0),
                          horizontal_flip=True, 
                          vertical_flip=True,
                          )

for file in os.listdir(dir):
  image = cv2.imread(os.path.join(dir, file))
  image = cv2.resize(image,(224, 224,))
  image = image.reshape((1,)+image.shape)
  i=0

  for batch in data_gen.flow(x=image, batch_size=1, save_to_dir='/img', 
                                  save_prefix='yes_' , save_format='png'):
      i += 1
      if i > 2:
          break

Этот код прекрасно работает, когда на вход подается только одно изображение, и генерирует три изображения.

В папке содержится 100 входных изображений, поэтому в общей сложности должно быть сгенерировано 300 изображений.

Но этот код не генерирует ровно 300 изображений. Вместо этого он генерирует 295 или 288 изображений при каждом выполнении этого кода.

В чем проблема, любая помощь?

Спасибо

...