Skip to content

Latest commit

 

History

History
45 lines (32 loc) · 1.28 KB

README.md

File metadata and controls

45 lines (32 loc) · 1.28 KB
Raw

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

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

top

Dependencies

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

Installation

To use this code, please do:

  1. Clone the repo:

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

  2. 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:

label

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

Noted that we do not implement Grabcut postprocessing here.