Cannot batch tensors with different shapes in component 0. First element had shape [224,224,3] and element 2 had shape [224,224,4]

I just want to know how can i load any custom dataset so I can process it to the model