ZVsion / rgbd_image_segmentation

Python implementation of "Interactive Segmentation on RGBD Images via Cue Selection"

Interactive-Segmentation-on-RGBD-Images-via-Cue-Selection

Python implementation of paper "Interactive Segmentation on RGBD Images via Cue Selection", CVPR 2016.

Dependencies

Python3.x (Tested with 3.5)
opencv
PyQt5
scikit-learn
PyMaxflow
Dijkstar

Installation

To use this code, please do:

0. Clone the repo:

   git clone https://github.com/ZVsion/rgbd_image_segmentation
   cd rgbd_image_segmentation

1. To try the demo, please run:

   python main.py

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:

For more details about this algorithm, please see in paper and code.

Noted that we do not implement Grabcut postprocessing here.