utils.colorize is dropping axes incorrectly

[roelofvandijk]

utils.colorize does not work properly.

Line 66 in utils.py is now: img = value[:, :, :3]

Should probably be: img = value.squeeze()