From 746a588b11c688001c77d7361d0143883e684723 Mon Sep 17 00:00:00 2001 From: dmnk1308 Date: Fri, 12 May 2023 14:41:25 +0200 Subject: [PATCH 1/4] fix typo and missing line --- dataset/default.py | 2 +- train/get_dataset.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/dataset/default.py b/dataset/default.py index 28bb0bd..7258364 100644 --- a/dataset/default.py +++ b/dataset/default.py @@ -26,7 +26,7 @@ def __init__(self, root_dir: str): def get_data_files(self): nifti_file_names = os.listdir(self.root_dir) folder_names = [os.path.join( - self.root_dir, nifti_file_name) for nifti_file_name in nifti_file_names if nifti_file_names.endsiwth('.nii')] + self.root_dir, nifti_file_name) for nifti_file_name in nifti_file_names if nifti_file_names.endswith('.nii')] return folder_names def __len__(self): diff --git a/train/get_dataset.py b/train/get_dataset.py index 5da9338..a3c4058 100644 --- a/train/get_dataset.py +++ b/train/get_dataset.py @@ -45,4 +45,5 @@ def get_dataset(cfg): val_dataset = DEFAULTDataset( root_dir=cfg.dataset.root_dir) sampler = None + return train_dataset, val_dataset, sampler raise ValueError(f'{cfg.dataset.name} Dataset is not available') From 0e997a9f801dec9882476efcc24dabc8692a4b31 Mon Sep 17 00:00:00 2001 From: dmnk1308 Date: Fri, 12 May 2023 15:06:34 +0200 Subject: [PATCH 2/4] append sys path --- train/train_vqgan.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/train/train_vqgan.py b/train/train_vqgan.py index d472321..8e78310 100644 --- a/train/train_vqgan.py +++ b/train/train_vqgan.py @@ -1,6 +1,8 @@ "Adapted from https://github.com/SongweiGe/TATS" import os +import sys +sys.path.append(os.path.split(os.getcwd())[0]) import pytorch_lightning as pl from pytorch_lightning.callbacks import ModelCheckpoint from torch.utils.data import DataLoader From f76d4f2cf3146c1552d46bc65c0c161cf7a2fa49 Mon Sep 17 00:00:00 2001 From: dmnk1308 Date: Fri, 12 May 2023 15:19:56 +0200 Subject: [PATCH 3/4] remove sys --- train/train_vqgan.py | 1 - 1 file changed, 1 deletion(-) diff --git a/train/train_vqgan.py b/train/train_vqgan.py index 8e78310..9ffc0d9 100644 --- a/train/train_vqgan.py +++ b/train/train_vqgan.py @@ -2,7 +2,6 @@ import os import sys -sys.path.append(os.path.split(os.getcwd())[0]) import pytorch_lightning as pl from pytorch_lightning.callbacks import ModelCheckpoint from torch.utils.data import DataLoader From 479a22b85635e1f28683db44e6e4ef31bbf4fa31 Mon Sep 17 00:00:00 2001 From: dmnk1308 Date: Fri, 12 May 2023 15:25:38 +0200 Subject: [PATCH 4/4] typo --- train/train_vqgan.py | 1 + 1 file changed, 1 insertion(+) diff --git a/train/train_vqgan.py b/train/train_vqgan.py index 9ffc0d9..3da2903 100644 --- a/train/train_vqgan.py +++ b/train/train_vqgan.py @@ -2,6 +2,7 @@ import os import sys +sys.path.append('/Users/dmnk/Documents/GitHub/segmentation_diffusion/medicaldiffusion') import pytorch_lightning as pl from pytorch_lightning.callbacks import ModelCheckpoint from torch.utils.data import DataLoader