generator.py 文件源码

def _generate_datasets(self):
        datasets = {'train': [], 'test': []}
        for dataset in self.datasets:
            print 'Generate dataset from {0}.'.format(dataset.name)
            # load dataset.
            dataset.load()
            # generate dataset
            for index in tqdm.trange(len(dataset), ascii=True):
                # get i-th data in the dataset.
                label, joint, image_file, image = dataset.get_data(index)
                # pad and crop image
                image, joint = self._pad_image(image, joint)
                image, joint = self._crop_image(image, joint)
                if not self._validate(joint):
                    continue
                # save the image
                image_path = self._save_image(dataset.name, image_file, image)
                # write database
                line = self._make_dataset_line(image_path, joint)
                datasets[label].append(line)
        return datasets