Hdooster

class ImageMaskGenerator(Sequence):
    def __init__(self, inputs, outputs, batch_size=32,
                 dim_after_preprocessing=None, shuffle=True,
                 input_channels=3, output_channels=1,
                 preprocessing_seq=None, augment_seq=None):
        self.inputs = inputs
        self.outputs = outputs
        self.input_channels = input_channels
        self.output_channels = output_channels
        self.dim_after_preprocessing = dim_after_preprocessing or self.inputs[0].shape[:2]