From 07ae2e998243619c86282fa53d6fe48bdac94d73 Mon Sep 17 00:00:00 2001 From: Jun-Yan Zhu Date: Thu, 18 Jul 2019 12:29:11 -0400 Subject: [PATCH] Update dataset directory for colorization --- data/colorization_dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/data/colorization_dataset.py b/data/colorization_dataset.py index 1c6d2c68e70..2616c61b649 100644 --- 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)