Upload model

lmdb_jpg.py

@@ -34,7 +34,7 @@ class JPGDataset(Dataset):
             'images': image,
         }
 
-def prepare_mimic_cxr_jpg_lmdb(mimic_iv_duckdb_path, mimic_cxr_jpg_path, mimic_cxr_jpg_lmdb_path, map_size_tb, num_workers=None):
+def prepare_mimic_cxr_jpg_lmdb(mimic_iv_duckdb_path, mimic_cxr_jpg_dir, mimic_cxr_jpg_lmdb_path, map_size_tb, num_workers=None):
     
     num_workers = num_workers if num_workers is not None else multiprocessing.cpu_count()
 
@@ -48,7 +48,7 @@ def prepare_mimic_cxr_jpg_lmdb(mimic_iv_duckdb_path, mimic_cxr_jpg_path, mimic_c
 
     print(f'Map size: {map_size}')
     
-    dataset = JPGDataset(df, mimic_cxr_jpg_path)
+    dataset = JPGDataset(df, mimic_cxr_jpg_dir)
     dataloader = DataLoader(
         dataset, 
         batch_size=num_workers,