def __init__(self, config): self._common_init(config) self.keypoint_num = 18 self.D_arch = config.D_arch if 'deepfashion' in config.dataset.lower(): if config.is_train: self.dataset_obj = deepfashion.get_split('train', config.data_path, data_name='DeepFashion') else: self.dataset_obj = deepfashion.get_split('test', config.data_path, data_name='DeepFashion') self.x, self.x_target, self.pose, self.pose_target, self.mask, self.mask_target = self._load_batch_pair_pose(self.dataset_obj)
def __init__(self, config): self._common_init(config) self.keypoint_num = 18 self.D_arch = config.D_arch # config.D_arch = DCGAN if ('deepfashion' in config.dataset.lower()) or ('df' in config.dataset.lower()): if config.is_train: self.dataset_obj = deepfashion.get_split('train', config.data_path, data_name='DeepFashion') else: self.dataset_obj = deepfashion.get_split('test', config.data_path, data_name='DeepFashion') self.x, self.x_target, self.pose, self.pose_target, self.mask, self.mask_target = self._load_batch_pair_pose(self.dataset_obj)