Deep Watershed Transform
Performs instance level segmentation detailed in the following paper:
Min Bai and Raquel Urtasun, Deep Watershed Transformation for Instance Segmentation, in CVPR 2017. Accessible at https://arxiv.org/abs/1611.08303.
This page is still under construction.
Developed and tested on Ubuntu 14.04 and 16.04.
- TensorFlow www.tensorflow.org
- Numpy, Scipy, and Skimage (sudo apt-get install python-numpy python-scipy python-skimage)
- Cityscapes images (www.cityscapes-dataset.com).
- Semantic Segmentation for input images. In our case, we used the output from PSPNet (by H. Zhao et al. https://github.com/hszhao/PSPNet). These are uint8 images with pixel-wise semantic labels encoded with 'trainIDs' defined by Cityscapes. For more information, visit https://github.com/mcordts/cityscapesScripts/blob/master/cityscapesscripts/helpers/labels.py
The model produces pixel-wise instance labels as a uint16 image with the same formatting as the Cityscapes instance segmentation challenge ground truth. In particular, each pixel is labeled as 'id' * 1000 + instance_id, where 'id' is as defined by Cityscapes (for more information, consult labels.py in the above link), and instance_id is an integer indexing the object instance.
Testing the Model
- Clone repository into dwt/.
- Download the model from www.cs.toronto.edu/~mbai/dwt_cityscapes_pspnet.mat and place into the "dwt/model" directory.
- run "cd E2E"
- run "python main.py"
- The results will be available in "dwt/example/output".
Training the Model
- Will be available soon.