Image classification: could not broadcast input array from shape into shape

Receiving this error trying to run a simple image classification example

  File "/Users/colinpetit/.pyenv/versions/3.6.5/lib/python3.6/site-packages/ludwig/features/image_feature.py", line 101, in add_feature_data
    data[feature['name']][i, :, :, :] = img
ValueError: could not broadcast input array from shape (743,620,3) into shape (383,442,4)

model_definition_file:

input_features:
  -
    name: img
    type: image
    encoder: stacked_cnn
output_features:
  -
    name: type
    type: category
training:
  epochs: 10

[w4nderlust]

Thanks @mattarderne for confirming it, will work on a patch that solves this issue, it will be pretty straightforward.

[w4nderlust]

I have to specify this better in the docs: Either your images are all of the same size, or you have to resize them to the same size by specifying the resize_method, width and height parameters as explained here. Let me know if this solves your issue.