Update dataset directory for colorization

EricKani · Jul 18, 2019 · 07ae2e9 · 07ae2e9
1 parent aec7ae3
commit 07ae2e9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/data/colorization_dataset.py b/data/colorization_dataset.py
@@ -36,7 +36,7 @@ def __init__(self, opt):
             opt (Option class) -- stores all the experiment flags; needs to be a subclass of BaseOptions
         """
         BaseDataset.__init__(self, opt)
-        self.dir = os.path.join(opt.dataroot)
+        self.dir = os.path.join(opt.dataroot, opt.phase)
         self.AB_paths = sorted(make_dataset(self.dir, opt.max_dataset_size))
         assert(opt.input_nc == 1 and opt.output_nc == 2 and opt.direction == 'AtoB')
         self.transform = get_transform(self.opt, convert=False)