def initialize(self, opt):
self.opt = opt
self.root = opt.dataroot
self.img_type = opt.img_type
self.dir_A = os.path.join(opt.dataroot, 'input')
self.dir_B = os.path.join(opt.dataroot, 'target')
self.dir_A_Map = os.path.join(opt.dataroot, 'seg_in')
self.dir_B_Map = os.path.join(opt.dataroot, 'seg_tar')
self.A_paths = make_dataset(self.dir_A)
self.A_paths = sorted(self.A_paths)
self.B_paths = make_dataset(self.dir_B)
self.B_paths = sorted(self.B_paths)
self.A_paths_map = make_dataset(self.dir_A_Map)
self.A_paths_map = sorted(self.A_paths_map)
self.B_paths_map = make_dataset(self.dir_B_Map)
self.B_paths_map = sorted(self.B_paths_map)
self.A_size = len(self.A_paths)
self.B_size = len(self.B_paths)
self.transform_type = get_transform_lab(opt)
self.transform_no = no_transform(opt)
def initialize(self, opt):
self.opt = opt
self.root = opt.dataroot
# Image - AVA
self.dir_A = os.path.join(opt.dataroot + 'image')
self.A_paths = make_dataset(self.dir_A)
self.A_paths = sorted(self.A_paths)
# Label - AVA
self.dir_BL = os.path.join(opt.dataroot + 'split_new')
self.BL_paths = make_dataset(self.dir_BL, mode='np')
self.BL_paths = sorted(self.BL_paths)
# Label - Anchor
self.dir_AL = os.path.join(opt.dataroot + 'split_above7')
self.AL_paths = make_dataset(self.dir_AL, mode='np')
self.AL_paths = sorted(self.AL_paths)
# Size
self.AVA_size = len(self.A_paths)
self.Anchor_size = len(self.AL_paths)
#self.AVA_size = self.AVA_size - 100
#self.transform = get_transform_vgg(opt)
self.transform = no_transform(opt)
ImageFile.LOAD_TRUNCATED_IMAGES = True
def initialize(self, opt):
self.opt = opt
self.root = opt.dataroot
self.dir_A = os.path.join(opt.dataroot, opt.phase + 'A')
self.A_paths = make_dataset(self.dir_A)
self.A_paths = sorted(self.A_paths)
self.A_size = len(self.A_paths)
self.transform_no = no_transform(opt)
def initialize(self, opt):
self.opt = opt
self.root = opt.dataroot
print(opt.video_folder)
self.dir_A = os.path.join(opt.dataroot, opt.video_folder)
self.A_paths = make_dataset(self.dir_A)
self.A_paths = sorted(self.A_paths)
self.A_size = len(self.A_paths)
self.transform_no = no_transform(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')
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_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_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.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.R_paths = sorted(self.R_paths)
self.transform = no_transform(opt)