Пример #1
0
    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)
Пример #2
0
    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)