segmentation.data
segmentation.data copied to clipboard
Processing deep-learning datasets (i.e. coco, pascal voc, cityscapes, pascal context)
DataProcessing
The project is for processing dataset, including Cityscapes and PASCAL VOC2012
Usage
Prerequisites:
- python 3
- numpy
- PIL
Visualization using multi-process
It consists of multi-process_visual & pallete
- run
multi-process_visual.pyfor converting gray predictions to colors. - it will use all the cpu are avaliable.
pallete.pyprovides palletes of different datasets, you can custom it yourself.
Converting index of Cityscapes labels
It consists of reverse_idx & cityscapes_labels
reverse_idx.pyprovides two functions for converting theidx.cityscapes_labelsis based on cityscapesScripts
Extra
contouris for computing the boundary maps used in pix2pixHD based on instance labels.scriptsis for coping desired images from files and generating lists of dataset (ie. w/ lst, w/o lst)
Update
-
coco2voc.pyconverts coco2017 labels, which are bigger than 1k pixels, to pascal voc format. This scripts based requires pycocotools and pytorch -
convert_pascal_context.pyconverts pascal context from 456 categories (.mat) to 59 categories (.png -- color & gray). I have listed themapping ids, you can also use funcsearch_map_idto generate it.
TODO
- [x] Converting scripts for PASCAL Context dataset
- [ ] Scripts for ADE20k