The project is for processing dataset, including Cityscapes and PASCAL VOC2012
- python 3
- numpy
- PIL
It consists of multi-process_visual
& pallete
- run
multi-process_visual.py
for converting gray predictions to colors. - it will use all the cpu are avaliable.
pallete.py
provides palletes of different datasets, you can custom it yourself.
It consists of reverse_idx
& cityscapes_labels
reverse_idx.py
provides two functions for converting theidx
.cityscapes_labels
is based on cityscapesScripts
contour
is for computing the boundary maps used in pix2pixHD based on instance labels.scripts
is for coping desired images from files and generating lists of dataset (ie. w/ lst, w/o lst)
-
coco2voc.py
converts coco2017 labels, which are bigger than 1k pixels, to pascal voc format. This scripts based requires pycocotools and pytorch -
convert_pascal_context.py
converts pascal context from 456 categories (.mat) to 59 categories (.png -- color & gray). I have listed themapping ids
, you can also use funcsearch_map_id
to generate it.
- Converting scripts for PASCAL Context dataset
- Scripts for ADE20k