| import argparse | |
| from datasets import load_dataset | |
| from tqdm import tqdm | |
| # choose a dataset | |
| available_datasets = ["apple2orange", "summer2winter_yosemite", "horse2zebra", "monet2photo", "cezanne2photo", "ukiyoe2photo", "vangogh2photo", "maps", "cityscapes", "facades", "iphone2dslr_flower", "ae_photos", "grumpifycat"] | |
| def upload_dataset(dataset_name): | |
| if dataset_name not in available_datasets: | |
| raise ValueError("Please choose one of the supported datasets:", available_datasets) | |
| # step 1: load dataset | |
| dataset = load_dataset("imagefolder", data_files=f"https://people.eecs.berkeley.edu/~taesung_park/CycleGAN/datasets/{dataset_name}.zip") | |
| # step 2: push to hub | |
| dataset.push_to_hub(f"huggan/{dataset_name}") | |
| def main(): | |
| parser = argparse.ArgumentParser() | |
| parser.add_argument("--dataset", default="apple2orange", type=str, help="Dataset to upload") | |
| args = parser.parse_args() | |
| upload_dataset(args.dataset) | |
| if __name__ == "__main__": | |
| main() | |