So here is the sample of code I am working on PANDA Kaggle competition dataset. I have read the tiff images using skimage.mulitimage library. So after making data loaders here is the code snippet
train_dataset = TrainDataset(X_train, y_train, transform= None) 
valid_dataset = TrainDataset(X_valid, y_valid, transform= None) 
train_loader = DataLoader(train_dataset, batch_size=4, num_workers = 4)
valid_loader = DataLoader(valid_dataset, batch_size=4, num_workers = 4)


So here in this image, you can see the shape of the image and their ground truth values.