Input images sometimes have different dimensions.

[madhavkhoslaa]

A png might not always have a 4 channel image and a tif might now have a 3 channel input. Assumptions taken like these in dataloader break the code.