()
| 75 | |
| 76 | |
| 77 | def parse_args(): |
| 78 | parser = argparse.ArgumentParser("", add_help=False) |
| 79 | parser.add_argument("--class_prompt", help="text prompt to retrieve images", required=True, type=str) |
| 80 | parser.add_argument("--class_data_dir", help="path to save images", required=True, type=str) |
| 81 | parser.add_argument("--num_class_images", help="number of images to download", default=200, type=int) |
| 82 | return parser.parse_args() |
| 83 | |
| 84 | |
| 85 | if __name__ == "__main__": |