datadings.sets.Cityscapes_write module

usage: Cityscapes_write.py [-h] [-o PATH] [-y] [-s] [–shuffle {yes,no}]

[-t 0-2] INDIR

Create CityScapes data set files.

This tool will look for the following files in the input directory:

  • disparity_trainvaltest.zip

  • gtFine_trainvaltest.zip

  • leftImg8bit_trainvaltest.zip

Note

Registration is required to download this dataset. Please visit the website to download it.

Important

Samples have the following keys:

  • "key"

  • "image"

  • "label_image"

  • "disparity_image"

Positional arguments

INDIR Directory that contains dataset source files.

options:
-h, --help

show this help message and exit

-o PATH, --outdir PATH

Output directory. Defaults to indir.

-y, --no-confirm

Don’t require user interaction.

-s, --skip-verification

Skip verification of source files.

–shuffle {yes,no} Write samples in random order. (default: yes) -t 0-2, –threads 0-2

Number of threads for conversion. 0 uses all available CPUs (default 1).

datadings.sets.Cityscapes_write.get_keys(leftzip, split)[source]
datadings.sets.Cityscapes_write.main()[source]
datadings.sets.Cityscapes_write.write_set(outdir, split, gen, total)[source]
datadings.sets.Cityscapes_write.write_sets(files, outdir, args)[source]
datadings.sets.Cityscapes_write.yield_samples(keys, leftzip, disparityzip, gtzip)[source]