Data augmentation to be improved.

[nuanyang123]

Thanks for your work. But there exists something to be improved in the part of data augmentation, or specifically, the crop/pad part of the code. Your code only considered the comparison between img.shape[0] and c.size_train[0] but did not consider the comparison between img.shape[1] and c.size_train[1].