From 0892d5fe994f5ac1e23381fb08540610145037e9 Mon Sep 17 00:00:00 2001 From: James Betker Date: Sun, 26 Jul 2020 22:48:35 -0600 Subject: [PATCH] LQGT_dataset gan debug --- codes/data/LQGT_dataset.py | 1 + 1 file changed, 1 insertion(+) diff --git a/codes/data/LQGT_dataset.py b/codes/data/LQGT_dataset.py index 183da765..5cc44b07 100644 --- a/codes/data/LQGT_dataset.py +++ b/codes/data/LQGT_dataset.py @@ -50,6 +50,7 @@ class LQGTDataset(data.Dataset): # LR and HR do not need to be paired. if 'dataroot_GAN' in opt.keys(): self.paths_GAN, self.sizes_GAN = util.get_image_paths(self.data_type, opt['dataroot_GAN']) + print('loaded %i images for use in training GAN only.' % (self.sizes_GAN,)) assert self.paths_GT, 'Error: GT path is empty.' if self.paths_LQ and self.paths_GT: