How to use __getitem__ to see dataset images' sizes

Hi,

I want to check whether train and test images (w and H) are in same size or not.
How can I load train/test data in getitem to achieve this purpose?

Thanks!