(self, tmpdir, config)
| 522 | _SPLIT_TO_IDX = dict(train=0, valid=1, test=2) |
| 523 | |
| 524 | def inject_fake_data(self, tmpdir, config): |
| 525 | base_folder = pathlib.Path(tmpdir) / "celeba" |
| 526 | os.makedirs(base_folder) |
| 527 | |
| 528 | num_images, num_images_per_split = self._create_split_txt(base_folder) |
| 529 | |
| 530 | datasets_utils.create_image_folder( |
| 531 | base_folder, "img_align_celeba", lambda idx: f"{idx + 1:06d}.jpg", num_images |
| 532 | ) |
| 533 | attr_names = self._create_attr_txt(base_folder, num_images) |
| 534 | self._create_identity_txt(base_folder, num_images) |
| 535 | self._create_bbox_txt(base_folder, num_images) |
| 536 | self._create_landmarks_txt(base_folder, num_images) |
| 537 | |
| 538 | num_samples = num_images_per_split.get(config["split"], 0) if isinstance(config["split"], str) else 0 |
| 539 | return dict(num_examples=num_samples, attr_names=attr_names) |
| 540 | |
| 541 | def _create_split_txt(self, root): |
| 542 | num_images_per_split = dict(train=4, valid=3, test=2) |
nothing calls this directly
no test coverage detected