datadings.sets.ADE20k_write module

usage: ADE20k_write.py [-h] [-o PATH] [-y] [-s] [–shuffle {yes,no}] [-t 0-2]

[–calculate-weights] [–scenelabels] INDIR

Create ADE20k data set files.

This tool will look for the following files in the input directory and download them if necessary:

  • ADE20K_2016_07_26.zip

See also

http://groups.csail.mit.edu/vision/datasets/ADE20K

Important

Samples have the following keys:

  • "key"

  • "image"

  • "label"

  • "label_image"

  • "parts_images"

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

--calculate-weights

Calculate median-frequency class weights.

--scenelabels

extract list of scene labels

datadings.sets.ADE20k_write.array_to_image(array, format, dtype, mode, **kwargs)[source]
datadings.sets.ADE20k_write.array_to_png16(array)[source]
datadings.sets.ADE20k_write.calculate_weights(files, outdir)[source]
datadings.sets.ADE20k_write.extract_scenelabels(files, outdir)[source]
datadings.sets.ADE20k_write.imagedata_to_segpng(data)[source]
datadings.sets.ADE20k_write.instance_map(im)[source]
datadings.sets.ADE20k_write.load_index(imagezip, *keys)[source]
datadings.sets.ADE20k_write.main()[source]
datadings.sets.ADE20k_write.segmentation_map(im)[source]
datadings.sets.ADE20k_write.write_set(imagezip, outdir, split, scenelabels, args)[source]
datadings.sets.ADE20k_write.write_sets(files, outdir, args)[source]
datadings.sets.ADE20k_write.yield_images(names)[source]