3

画像マスクがラベルの代わりに使用されている場合に image_datagen.flow_from_directory() を設定する方法を示す Keras ドキュメントのサンプルコードの変更を使用しようとしていました (画像セグメンテーションのために、クラスを予測しています)各ピクセル)。

ちなみに、各画像のカラー チャネルからすべてのトレーニング画像の各カラー チャネルの平均を差し引くために、featurewise_center = True を設定して、トレーニング セット全体で各カラー チャネルの平均が 0 になるようにします。これを達成する方法ではありません。

とにかく、エラーを生成したコードは次のとおりです。

image_datagen = ImageDataGenerator(featurewise_center = True)
mask_datagen = ImageDataGenerator()

image_generator = image_datagen.flow_from_directory(
    '/home/icg/Martin/train_data_graz/images_rect_r640x360',
    class_mode = None,
    batch_size = 1,
    seed = 123)

mask_generator = mask_datagen.flow_from_directory(
    '/home/icg/Martin/train_data_graz/labels_rect_r640x360',
    class_mode = None,
    batch_size = 1,
    seed = 123)

# combine generators into one which yields image and masks
train_generator = zip(image_generator, mask_generator)

model.fit_generator(
    train_generator,
    steps_per_epoch = 1000,
    epochs = 100)

エラーメッセージは次のとおりです。 

Found 0 images belonging to 0 classes.
Found 0 images belonging to 0 classes.
Traceback (most recent call last):
  File "FCN_VGG16.py", line 178, in <module>
    train_generator = zip(image_generator, mask_generator)
  File "/home/icg/rafa/local/lib/python2.7/site-packages/keras/preprocessing/image.py", line 1026, in next
    index_array, current_index, current_batch_size = next(self.index_generator)
  File "/home/icg/rafa/local/lib/python2.7/site-packages/keras/preprocessing/image.py", line 720, in _flow_index
    current_index = (self.batch_index * batch_size) % n
ZeroDivisionError: integer division or modulo by zero

なんらかの理由で n = 0 です。これが発生する理由はありますか?

4

1 に答える 1