apes / docs /dataset-tool-help.txt
tkarras's picture
Documentation
a6f45eb
raw
history blame
2.31 kB
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.