Skip to content
Python implementation of "Interactive Segmentation on RGBD Images via Cue Selection"
Branch: master
Clone or download
Latest commit 389191c Jun 10, 2018
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
Imgs Completed Version Jun 10, 2018
Results Completed Version Jun 10, 2018
doc Completed Version Jun 10, 2018
.gitignore Completed Version Jun 10, 2018
GraphMaker.py Completed Version Jun 10, 2018
README.md Completed Version Jun 10, 2018
main.py Completed Version Jun 10, 2018

README.md

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

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

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, therefore, our results are not as good as original paper.

You can’t perform that action at this time.