ImprintLab · halleewong · Oct 22, 2024 · Oct 22, 2024 · Oct 22, 2024
diff --git a/.gitignore b/.gitignore
@@ -0,0 +1,2 @@
+*__pycache__*
+__pycache__*
diff --git a/__pycache__/cfg.cpython-311.pyc b/__pycache__/cfg.cpython-311.pyc
diff --git a/__pycache__/dataset.cpython-311.pyc b/__pycache__/dataset.cpython-311.pyc
diff --git a/__pycache__/function.cpython-311.pyc b/__pycache__/function.cpython-311.pyc
diff --git a/__pycache__/utils.cpython-311.pyc b/__pycache__/utils.cpython-311.pyc
diff --git a/conf/__pycache__/__init__.cpython-311.pyc b/conf/__pycache__/__init__.cpython-311.pyc
diff --git a/conf/__pycache__/global_settings.cpython-311.pyc b/conf/__pycache__/global_settings.cpython-311.pyc
diff --git a/dataset.py b/dataset.py
@@ -22,9 +22,9 @@ class ISIC2016(Dataset):
     def __init__(self, args, data_path , transform = None, transform_msk = None, mode = 'Training',prompt = 'click', plane = False):
 
         df = pd.read_csv(os.path.join(data_path, 'ISBI2016_ISIC_Part3B_' + mode + '_GroundTruth.csv'), encoding='gbk')
-        self.name_list = df.iloc[:,1].tolist()
-        self.label_list = df.iloc[:,2].tolist()
-        self.data_path = data_path
+        self.name_list = df.iloc[:,0].tolist()
+        self.label_list = df.iloc[:,1].tolist()
+        self.data_path = os.path.join(data_path, 'ISBI2016_ISIC_Part3B_' + mode + "_Data")
         self.mode = mode
         self.prompt = prompt
         self.img_size = args.image_size
@@ -40,10 +40,10 @@ def __getitem__(self, index):
         point_label = 1
 
         """Get the images"""
-        name = self.name_list[index]
+        name = self.name_list[index] + ".jpg"
         img_path = os.path.join(self.data_path, name)
 
-        mask_name = self.label_list[index]
+        mask_name = self.name_list[index] + "_Segmentation.png"
         msk_path = os.path.join(self.data_path, mask_name)
 
         img = Image.open(img_path).convert('RGB')
@@ -60,11 +60,9 @@ def __getitem__(self, index):
             img = self.transform(img)
             torch.set_rng_state(state)
 
-
             if self.transform_msk:
                 mask = self.transform_msk(mask)
 
-
         name = name.split('/')[-1].split(".jpg")[0]
         image_meta_dict = {'filename_or_obj':name}
         return {

diff --git a/models/__pycache__/discriminator.cpython-311.pyc b/models/__pycache__/discriminator.cpython-311.pyc
diff --git a/models/oneprompt/__pycache__/__init__.cpython-311.pyc b/models/oneprompt/__pycache__/__init__.cpython-311.pyc
diff --git a/models/oneprompt/__pycache__/automatic_mask_generator.cpython-311.pyc b/models/oneprompt/__pycache__/automatic_mask_generator.cpython-311.pyc
diff --git a/models/oneprompt/__pycache__/build_one.cpython-311.pyc b/models/oneprompt/__pycache__/build_one.cpython-311.pyc
diff --git a/models/oneprompt/__pycache__/build_oneprompt.cpython-311.pyc b/models/oneprompt/__pycache__/build_oneprompt.cpython-311.pyc
diff --git a/models/oneprompt/__pycache__/build_sam.cpython-311.pyc b/models/oneprompt/__pycache__/build_sam.cpython-311.pyc
diff --git a/models/oneprompt/__pycache__/predictor.cpython-311.pyc b/models/oneprompt/__pycache__/predictor.cpython-311.pyc
diff --git a/models/oneprompt/modeling/__pycache__/__init__.cpython-311.pyc b/models/oneprompt/modeling/__pycache__/__init__.cpython-311.pyc
diff --git a/models/oneprompt/modeling/__pycache__/common.cpython-311.pyc b/models/oneprompt/modeling/__pycache__/common.cpython-311.pyc
diff --git a/models/oneprompt/modeling/__pycache__/fp16_util.cpython-311.pyc b/models/oneprompt/modeling/__pycache__/fp16_util.cpython-311.pyc
diff --git a/models/oneprompt/modeling/__pycache__/image_encoder.cpython-311.pyc b/models/oneprompt/modeling/__pycache__/image_encoder.cpython-311.pyc
diff --git a/models/oneprompt/modeling/__pycache__/mask_decoder.cpython-311.pyc b/models/oneprompt/modeling/__pycache__/mask_decoder.cpython-311.pyc
diff --git a/models/oneprompt/modeling/__pycache__/modules.cpython-311.pyc b/models/oneprompt/modeling/__pycache__/modules.cpython-311.pyc
diff --git a/models/oneprompt/modeling/__pycache__/nn.cpython-311.pyc b/models/oneprompt/modeling/__pycache__/nn.cpython-311.pyc
diff --git a/models/oneprompt/modeling/__pycache__/oneprompt.cpython-311.pyc b/models/oneprompt/modeling/__pycache__/oneprompt.cpython-311.pyc
diff --git a/models/oneprompt/modeling/__pycache__/prompt_encoder.cpython-311.pyc b/models/oneprompt/modeling/__pycache__/prompt_encoder.cpython-311.pyc
diff --git a/models/oneprompt/modeling/__pycache__/sam.cpython-311.pyc b/models/oneprompt/modeling/__pycache__/sam.cpython-311.pyc
diff --git a/models/oneprompt/modeling/__pycache__/transformer.cpython-311.pyc b/models/oneprompt/modeling/__pycache__/transformer.cpython-311.pyc
diff --git a/models/oneprompt/modeling/__pycache__/utils.cpython-311.pyc b/models/oneprompt/modeling/__pycache__/utils.cpython-311.pyc
diff --git a/models/oneprompt/utils/__pycache__/__init__.cpython-311.pyc b/models/oneprompt/utils/__pycache__/__init__.cpython-311.pyc
diff --git a/models/oneprompt/utils/__pycache__/amg.cpython-311.pyc b/models/oneprompt/utils/__pycache__/amg.cpython-311.pyc
diff --git a/models/oneprompt/utils/__pycache__/transforms.cpython-311.pyc b/models/oneprompt/utils/__pycache__/transforms.cpython-311.pyc
diff --git a/pytorch_ssim/__pycache__/__init__.cpython-311.pyc b/pytorch_ssim/__pycache__/__init__.cpython-311.pyc