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. | |