Python implementation of paper "Interactive Segmentation on RGBD Images via Cue Selection".
Python3.x (Tested with 3.5) opencv PyQt5 scikit-learn PyMaxflow Dijkstar
To use this code, please do:
Clone the repo:
git clone https://github.com/ZVsion/rgbd_image_segmentation cd rgbd_image_segmentation
To try the demo, please run:
If installed correctly, the GUI should look like this:
You can add some foreground and background seeds in image, and then click
Segment Image. After a few seconds,
the result should look like this:
Each color's label is presented as follow:
Noted that we do not implement Grabcut postprocessing, therefore, our results are not as good as original paper.