()
| 499 | |
| 500 | |
| 501 | def main(): |
| 502 | try: |
| 503 | dash_index = sys.argv.index("--") |
| 504 | except ValueError as exc: |
| 505 | raise ValueError("arguments must be preceded by '--'") from exc |
| 506 | |
| 507 | raw_args = sys.argv[dash_index + 1 :] |
| 508 | parser = argparse.ArgumentParser() |
| 509 | parser.add_argument("--input_path", required=True, type=str) |
| 510 | parser.add_argument("--output_path", required=True, type=str) |
| 511 | parser.add_argument("--num_images", type=int, default=20) |
| 512 | parser.add_argument("--backend", type=str, default="BLENDER_EEVEE") |
| 513 | parser.add_argument("--light_mode", type=str, default="uniform") |
| 514 | parser.add_argument("--camera_pose", type=str, default="random") |
| 515 | parser.add_argument("--camera_dist_min", type=float, default=2.0) |
| 516 | parser.add_argument("--camera_dist_max", type=float, default=2.0) |
| 517 | parser.add_argument("--fast_mode", action="store_true") |
| 518 | args = parser.parse_args(raw_args) |
| 519 | |
| 520 | save_rendering_dataset( |
| 521 | input_path=args.input_path, |
| 522 | output_path=args.output_path, |
| 523 | num_images=args.num_images, |
| 524 | backend=args.backend, |
| 525 | light_mode=args.light_mode, |
| 526 | camera_pose=args.camera_pose, |
| 527 | camera_dist_min=args.camera_dist_min, |
| 528 | camera_dist_max=args.camera_dist_max, |
| 529 | fast_mode=args.fast_mode, |
| 530 | ) |
| 531 | |
| 532 | |
| 533 | main() |
no test coverage detected