datadings.sets.RIT18_write module

usage: RIT18_write.py [-h] [-o PATH] [-y] [-s] INDIR

Create RIT18 data set files.

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

  • rit18_data.mat

Warning

Images, labels, and masks are numpy arrays, not images!

Important

Samples have the following keys:

  • "key"

  • "image"

  • "label_image"

  • "mask"

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.

datadings.sets.RIT18_write.main()[source]
datadings.sets.RIT18_write.write(writer, img, labels, mask, filename)[source]
datadings.sets.RIT18_write.write_set(outdir, split, labels, data, args, crop_size)[source]
datadings.sets.RIT18_write.write_sets(files, outdir, args, crop_size=(384, 384))[source]
datadings.sets.RIT18_write.write_train(outdir, dataset, args)[source]