diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..ccdd459 --- /dev/null +++ b/.gitignore @@ -0,0 +1,2 @@ +*__pycache__* +__pycache__* diff --git a/__pycache__/cfg.cpython-311.pyc b/__pycache__/cfg.cpython-311.pyc deleted file mode 100644 index 8415cda..0000000 Binary files a/__pycache__/cfg.cpython-311.pyc and /dev/null differ diff --git a/__pycache__/dataset.cpython-311.pyc b/__pycache__/dataset.cpython-311.pyc deleted file mode 100644 index 3a0c8d7..0000000 Binary files a/__pycache__/dataset.cpython-311.pyc and /dev/null differ diff --git a/__pycache__/function.cpython-311.pyc b/__pycache__/function.cpython-311.pyc deleted file mode 100644 index 3ef576a..0000000 Binary files a/__pycache__/function.cpython-311.pyc and /dev/null differ diff --git a/__pycache__/utils.cpython-311.pyc b/__pycache__/utils.cpython-311.pyc deleted file mode 100644 index f71ec4b..0000000 Binary files a/__pycache__/utils.cpython-311.pyc and /dev/null differ diff --git a/conf/__pycache__/__init__.cpython-311.pyc b/conf/__pycache__/__init__.cpython-311.pyc deleted file mode 100644 index 76d8d61..0000000 Binary files a/conf/__pycache__/__init__.cpython-311.pyc and /dev/null differ diff --git a/conf/__pycache__/global_settings.cpython-311.pyc b/conf/__pycache__/global_settings.cpython-311.pyc deleted file mode 100644 index 912b468..0000000 Binary files a/conf/__pycache__/global_settings.cpython-311.pyc and /dev/null differ diff --git a/dataset.py b/dataset.py index a4b049c..ac42690 100644 --- 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 deleted file mode 100644 index 9d011bd..0000000 Binary files a/models/__pycache__/discriminator.cpython-311.pyc and /dev/null differ diff --git a/models/oneprompt/__pycache__/__init__.cpython-311.pyc b/models/oneprompt/__pycache__/__init__.cpython-311.pyc deleted file mode 100644 index af1f6f4..0000000 Binary files a/models/oneprompt/__pycache__/__init__.cpython-311.pyc and /dev/null differ diff --git a/models/oneprompt/__pycache__/automatic_mask_generator.cpython-311.pyc b/models/oneprompt/__pycache__/automatic_mask_generator.cpython-311.pyc deleted file mode 100644 index 542ea41..0000000 Binary files a/models/oneprompt/__pycache__/automatic_mask_generator.cpython-311.pyc and /dev/null differ diff --git a/models/oneprompt/__pycache__/build_one.cpython-311.pyc b/models/oneprompt/__pycache__/build_one.cpython-311.pyc deleted file mode 100644 index 5c06876..0000000 Binary files a/models/oneprompt/__pycache__/build_one.cpython-311.pyc and /dev/null differ diff --git a/models/oneprompt/__pycache__/build_oneprompt.cpython-311.pyc b/models/oneprompt/__pycache__/build_oneprompt.cpython-311.pyc deleted file mode 100644 index 80919fb..0000000 Binary files a/models/oneprompt/__pycache__/build_oneprompt.cpython-311.pyc and /dev/null differ diff --git a/models/oneprompt/__pycache__/build_sam.cpython-311.pyc b/models/oneprompt/__pycache__/build_sam.cpython-311.pyc deleted file mode 100644 index 30d650e..0000000 Binary files a/models/oneprompt/__pycache__/build_sam.cpython-311.pyc and /dev/null differ diff --git a/models/oneprompt/__pycache__/predictor.cpython-311.pyc b/models/oneprompt/__pycache__/predictor.cpython-311.pyc deleted file mode 100644 index 31be16d..0000000 Binary files a/models/oneprompt/__pycache__/predictor.cpython-311.pyc and /dev/null differ diff --git a/models/oneprompt/modeling/__pycache__/__init__.cpython-311.pyc b/models/oneprompt/modeling/__pycache__/__init__.cpython-311.pyc deleted file mode 100644 index f333a01..0000000 Binary files a/models/oneprompt/modeling/__pycache__/__init__.cpython-311.pyc and /dev/null differ diff --git a/models/oneprompt/modeling/__pycache__/common.cpython-311.pyc b/models/oneprompt/modeling/__pycache__/common.cpython-311.pyc deleted file mode 100644 index f654513..0000000 Binary files a/models/oneprompt/modeling/__pycache__/common.cpython-311.pyc and /dev/null differ diff --git a/models/oneprompt/modeling/__pycache__/fp16_util.cpython-311.pyc b/models/oneprompt/modeling/__pycache__/fp16_util.cpython-311.pyc deleted file mode 100644 index bb1e11c..0000000 Binary files a/models/oneprompt/modeling/__pycache__/fp16_util.cpython-311.pyc and /dev/null differ diff --git a/models/oneprompt/modeling/__pycache__/image_encoder.cpython-311.pyc b/models/oneprompt/modeling/__pycache__/image_encoder.cpython-311.pyc deleted file mode 100644 index 46791a5..0000000 Binary files a/models/oneprompt/modeling/__pycache__/image_encoder.cpython-311.pyc and /dev/null differ diff --git a/models/oneprompt/modeling/__pycache__/mask_decoder.cpython-311.pyc b/models/oneprompt/modeling/__pycache__/mask_decoder.cpython-311.pyc deleted file mode 100644 index dfcd9c3..0000000 Binary files a/models/oneprompt/modeling/__pycache__/mask_decoder.cpython-311.pyc and /dev/null differ diff --git a/models/oneprompt/modeling/__pycache__/modules.cpython-311.pyc b/models/oneprompt/modeling/__pycache__/modules.cpython-311.pyc deleted file mode 100644 index b748985..0000000 Binary files a/models/oneprompt/modeling/__pycache__/modules.cpython-311.pyc and /dev/null differ diff --git a/models/oneprompt/modeling/__pycache__/nn.cpython-311.pyc b/models/oneprompt/modeling/__pycache__/nn.cpython-311.pyc deleted file mode 100644 index b5d4897..0000000 Binary files a/models/oneprompt/modeling/__pycache__/nn.cpython-311.pyc and /dev/null differ diff --git a/models/oneprompt/modeling/__pycache__/oneprompt.cpython-311.pyc b/models/oneprompt/modeling/__pycache__/oneprompt.cpython-311.pyc deleted file mode 100644 index 7198a38..0000000 Binary files a/models/oneprompt/modeling/__pycache__/oneprompt.cpython-311.pyc and /dev/null differ diff --git a/models/oneprompt/modeling/__pycache__/prompt_encoder.cpython-311.pyc b/models/oneprompt/modeling/__pycache__/prompt_encoder.cpython-311.pyc deleted file mode 100644 index 54c19db..0000000 Binary files a/models/oneprompt/modeling/__pycache__/prompt_encoder.cpython-311.pyc and /dev/null differ diff --git a/models/oneprompt/modeling/__pycache__/sam.cpython-311.pyc b/models/oneprompt/modeling/__pycache__/sam.cpython-311.pyc deleted file mode 100644 index 4476e04..0000000 Binary files a/models/oneprompt/modeling/__pycache__/sam.cpython-311.pyc and /dev/null differ diff --git a/models/oneprompt/modeling/__pycache__/transformer.cpython-311.pyc b/models/oneprompt/modeling/__pycache__/transformer.cpython-311.pyc deleted file mode 100644 index 4aec8a7..0000000 Binary files a/models/oneprompt/modeling/__pycache__/transformer.cpython-311.pyc and /dev/null differ diff --git a/models/oneprompt/modeling/__pycache__/utils.cpython-311.pyc b/models/oneprompt/modeling/__pycache__/utils.cpython-311.pyc deleted file mode 100644 index 32ee4fd..0000000 Binary files a/models/oneprompt/modeling/__pycache__/utils.cpython-311.pyc and /dev/null differ diff --git a/models/oneprompt/utils/__pycache__/__init__.cpython-311.pyc b/models/oneprompt/utils/__pycache__/__init__.cpython-311.pyc deleted file mode 100644 index a0542fc..0000000 Binary files a/models/oneprompt/utils/__pycache__/__init__.cpython-311.pyc and /dev/null differ diff --git a/models/oneprompt/utils/__pycache__/amg.cpython-311.pyc b/models/oneprompt/utils/__pycache__/amg.cpython-311.pyc deleted file mode 100644 index 0b323fd..0000000 Binary files a/models/oneprompt/utils/__pycache__/amg.cpython-311.pyc and /dev/null differ diff --git a/models/oneprompt/utils/__pycache__/transforms.cpython-311.pyc b/models/oneprompt/utils/__pycache__/transforms.cpython-311.pyc deleted file mode 100644 index 0b04e37..0000000 Binary files a/models/oneprompt/utils/__pycache__/transforms.cpython-311.pyc and /dev/null differ diff --git a/pytorch_ssim/__pycache__/__init__.cpython-311.pyc b/pytorch_ssim/__pycache__/__init__.cpython-311.pyc deleted file mode 100644 index b3d0322..0000000 Binary files a/pytorch_ssim/__pycache__/__init__.cpython-311.pyc and /dev/null differ