Upload model

dataset.py

@@ -1,12 +1,9 @@
 import os
-import struct
 
-import lmdb
-import numpy as np
 import pandas as pd
 import torch
 from torch.utils.data import Dataset
-from torchvision.io import decode_image, read_image
+from torchvision.io import read_image
 
 from tools.utils import mimic_cxr_image_path
 
@@ -70,7 +67,7 @@ class StudyIDEDStayIDSubset(Dataset):
 
         assert self.extension == 'jpg' or self.extension == 'dcm'
 
-        if self.dataset_dir is not None and self.images_rocksdb_path is None:
+        if self.dataset_dir is not None:
             if self.extension == 'jpg':
                 if 'physionet.org/files/mimic-cxr-jpg/2.0.0/files' not in self.dataset_dir:
                     self.dataset_dir = os.path.join(self.dataset_dir, 'physionet.org/files/mimic-cxr-jpg/2.0.0/files')