def initialize(self, opt):
self.opt = opt
self.root = opt.dataroot
if opt.phase == 'train':
self.dir_A = os.path.join(opt.dataroot+'trainB_exA_resized')
self.dir_B = os.path.join(opt.dataroot+'trainB_exB_resized')
self.dir_C = os.path.join(opt.dataroot+'trainB_exC_resized')
self.dir_D = os.path.join(opt.dataroot+'trainB_exD_resized')
self.dir_E = os.path.join(opt.dataroot+'trainB_exE_resized')
self.dir_M = os.path.join(opt.dataroot+'trainB_exA_resized') ##### M
self.dir_N = os.path.join(opt.dataroot+'trainB_exB_resized') ##### N
#self.dir_A = os.path.join(opt.dataroot+'Jpeg/expertA_test')
#self.dir_B = os.path.join(opt.dataroot+'Jpeg/expertB_test')
#self.dir_C = os.path.join(opt.dataroot+'Jpeg/expertC_test')
#self.dir_D = os.path.join(opt.dataroot+'Jpeg/expertD_test')
#self.dir_E = os.path.join(opt.dataroot+'Jpeg/expertE_test')
self.dir_R = os.path.join(opt.dataroot+'trainA')
if opt.phase == 'test':
self.dir_A = os.path.join(opt.dataroot+'testB_exA_resized')
self.dir_B = os.path.join(opt.dataroot+'testB_exB_resized')
self.dir_C = os.path.join(opt.dataroot+'testB_exC_resized')
self.dir_D = os.path.join(opt.dataroot+'testB_exD_resized')
self.dir_E = os.path.join(opt.dataroot+'testB_exE_resized')
self.dir_M = os.path.join(opt.dataroot+'testB_exA_resized') ##### M
self.dir_N = os.path.join(opt.dataroot+'testB_exB_resized') ##### N
#self.dir_A = os.path.join(opt.dataroot+'Jpeg/expertA_test')
#self.dir_B = os.path.join(opt.dataroot+'Jpeg/expertB_test')
#self.dir_C = os.path.join(opt.dataroot+'Jpeg/expertC_test')
#self.dir_D = os.path.join(opt.dataroot+'Jpeg/expertD_test')
#self.dir_E = os.path.join(opt.dataroot+'Jpeg/expertE_test')
self.dir_R = os.path.join(opt.dataroot+'testA')
self.A_paths = make_dataset(self.dir_A)
self.B_paths = make_dataset(self.dir_B)
self.C_paths = make_dataset(self.dir_C)
self.D_paths = make_dataset(self.dir_D)
self.E_paths = make_dataset(self.dir_E)
self.M_paths = make_dataset(self.dir_M)
self.N_paths = make_dataset(self.dir_N)
self.R_paths = make_dataset(self.dir_R)
self.A_paths = sorted(self.A_paths)
self.B_paths = sorted(self.B_paths)
self.C_paths = sorted(self.C_paths)
self.D_paths = sorted(self.D_paths)
self.E_paths = sorted(self.E_paths)
self.M_paths = sorted(self.M_paths)
self.N_paths = sorted(self.N_paths)
self.R_paths = sorted(self.R_paths)
self.M_size = len(self.M_paths)
self.N_size = len(self.N_paths)
self.transform = get_transform_vgg(opt)
def initialize(self, opt):
self.opt = opt
self.root = opt.dataroot
if opt.phase == 'test':
self.dir_Single = os.path.join(opt.dataroot + 'testB_singledata')
self.Single_paths = make_dataset(self.dir_Single)
self.Single_paths = sorted(self.Single_paths)
self.Single_size = len(self.Single_paths)
self.transform = get_transform_vgg(opt)
def initialize(self, opt):
self.opt = opt
self.root = opt.dataroot
if opt.phase == 'train':
self.dir_A = os.path.join(opt.dataroot + 'trainB_exA_resized')
self.dir_B = os.path.join(opt.dataroot + 'trainB_exB_resized')
if opt.phase == 'test':
self.dir_A = os.path.join(opt.dataroot + 'testB_exA_resized')
self.dir_B = os.path.join(opt.dataroot + 'testB_exB_resized')
self.A_paths = make_dataset(self.dir_A)
self.B_paths = make_dataset(self.dir_B)
self.A_paths = sorted(self.A_paths)
self.B_paths = sorted(self.B_paths)
self.transform = get_transform_vgg(opt)