Update dataloader.py

NYUMedML · Aug 5, 2022 · b58070c · b58070c
1 parent ebe6d00
commit b58070c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/dataset/dataloader.py b/dataset/dataloader.py
@@ -3,6 +3,8 @@
 import torch
 from PIL import Image
 from torch.utils.data import Dataset
+from torchvision import transforms
+import cv2
 
 class TCGA_CPTAC_Dataset(Dataset):
     def __init__(self, cptac_dir, tcga_dir, split_dir, transform=None, mode='train', batch_slide_num=4, batch_size=128):