Spaces:
Running
Running
| Usage: dataset_tool.py [OPTIONS] | |
| Convert an image dataset into a dataset archive usable with StyleGAN2 ADA | |
| PyTorch. | |
| The input dataset format is guessed from the --source argument: | |
| --source *_lmdb/ - Load LSUN dataset | |
| --source cifar-10-python.tar.gz - Load CIFAR-10 dataset | |
| --source path/ - Recursively load all images from path/ | |
| --source dataset.zip - Recursively load all images from dataset.zip | |
| The output dataset format can be either an image folder or a zip archive. | |
| Specifying the output format and path: | |
| --dest /path/to/dir - Save output files under /path/to/dir | |
| --dest /path/to/dataset.zip - Save output files into /path/to/dataset.zip archive | |
| Images within the dataset archive will be stored as uncompressed PNG. | |
| Image scale/crop and resolution requirements: | |
| Output images must be square-shaped and they must all have the same power- | |
| of-two dimensions. | |
| To scale arbitrary input image size to a specific width and height, use | |
| the --width and --height options. Output resolution will be either the | |
| original input resolution (if --width/--height was not specified) or the | |
| one specified with --width/height. | |
| Use the --transform=center-crop or --transform=center-crop-wide options to | |
| apply a center crop transform on the input image. These options should be | |
| used with the --width and --height options. For example: | |
| python dataset_tool.py --source LSUN/raw/cat_lmdb --dest /tmp/lsun_cat \ | |
| --transform=center-crop-wide --width 512 --height=384 | |
| Options: | |
| --source PATH Directory or archive name for input dataset | |
| [required] | |
| --dest PATH Output directory or archive name for output | |
| dataset [required] | |
| --max-images INTEGER Output only up to `max-images` images | |
| --resize-filter [box|lanczos] Filter to use when resizing images for | |
| output resolution [default: lanczos] | |
| --transform [center-crop|center-crop-wide] | |
| Input crop/resize mode | |
| --width INTEGER Output width | |
| --height INTEGER Output height | |
| --help Show this message and exit. | |