ValueError: Error when checking input: expected input_1 to have 4 dimensions, but got array with shape (250, 250, 3) about keras-fcn

[theladyjaye]

Ok I refactored my generator after looking at your for voc2011:

    img_input = img_to_array(load_img(path_input))
    img_target = img_to_array(load_img(path_target))
    batch_input = np.zeros((1,) + (250, 250, 3), dtype=K.floatx())
    batch_target = np.zeros((1,) + (250, 250, 3), dtype=K.floatx())

    batch_input[0] = img_input
    batch_target[0] = img_target

    yield (batch_input, batch_target)

And that did the trick.

Also, question for my own ignorance:

https://github.com/JihongJu/keras-fcn/blob/master/keras_fcn/models.py#L61
https://github.com/JihongJu/keras-fcn/blob/master/keras_fcn/models.py#L98

On those 2 lines the classes are hard coded against 21 instead of taking the classes value from the passed in args. Is that intentional? I know there are default 21 classes, but in the case where you are doing something custom, wouldn't it make sense to allow that to be configured?

[JihongJu]

Hi @aventurella , thank you for the suggestion! Those two places indeed should be classes passed from the argument instead of hard coded 21.

[JihongJu]

It is now fixed. Thank you! @aventurella