Spaces:

deepinv
/

denoising

Sleeping

msong97 commited on 13 days ago

Commit

a5ab072

1 Parent(s): a0ca102

Typo in directory name of dataset

Files changed (1) hide show

factories.py CHANGED Viewed

@@ -447,7 +447,7 @@ class EvalDataset(torch.utils.data.Dataset):
                                                 transform=self.transform,
                                                 preprocess=False)
         elif self.name == "CT":
-            self.root = 'img_samples/LIDC_IDRI_samples'
             self.transform = None
             self.dataset = Preprocessed_LIDCIDRI(root=self.root,
                                                  transform=self.transform)

                                                 transform=self.transform,
                                                 preprocess=False)
         elif self.name == "CT":
+            self.root = 'img_samples/LIDC-IDRI_samples'
             self.transform = None
             self.dataset = Preprocessed_LIDCIDRI(root=self.root,
                                                  transform=self.transform)