IsabelaBB / iDISF

iDISF implementation and a tool for interactive image segmentation.

Interactive Dynamic and Iterative Spanning Forest (iDISF)

This repository contains a tool for interactive image segmentation and an iDISF implementation.

Interactive DISF (iDISF)

From an image labeled with scribbles, it performs an oversampling of background seeds and, through the iterations, generates superpixels and removes those considered as irrelevant. In this work, removal strategies by class and relevance are used. In removing by class the algorithm performs a desired number of iterations, and in removing by relevance the algorithm stops when reaching the desired amount of superpixels.

Languages supported

C/C++ (Implementation) and Python3 (Wrapper)

Requirements

The following tools are required in order to start the installation.

• OpenMP
• Python 3

Installing

The following python libraries are required to compile the project.

1. Package to build our iDISF python module: pip3 install scikit-build cmake
2. Install Tkinter libraries for python3 to run the interface: sudo apt-get install python3-tk; python3 -m pip install git+https://github.com/RedFantom/ttkthemes
3. Install other common python libraries: pip3 install -r requirements.txt

Compiling and cleaning

• To compile all files: make
• For removing all generated files from source: make clean

Running

In this project, there are two demo files, one for each language supported (i.e., C and Python3). After compiling and assuring the generation of the necessary files, one can execute each demo within its own environment.

iDISF code only

• The complete iDISF project can be executed by ./bin/DISF_demo
• For the command options, run ./bin/DISF_demo --help
• A demo example from iDISF in python3: python3 DISF_demo.py

Interface

• Run the user interface with the follow command: python3 iDISF.py
• Our interface includes iDISF and Watershed segmentation. For Watershed, we use higra package.

Cite

If this work was useful for your research, please cite our paper:

@InProceedings{barcelos2021towards,
  title={Towards Interactive Image Segmentation by Dynamic and Iterative Spanning Forest},
  author={Barcelos, Isabela Borlido and Bel{\'e}m, Felipe and Miranda, Paulo and Falc{\~a}o, Alexandre Xavier and do Patroc{\'\i}nio, Zenilton KG and Guimar{\~a}es, Silvio Jamil F},
  booktitle={International Conference on Discrete Geometry and Mathematical Morphology},
  publisher={Springer International Publishing},
  pages={351--364},
  year={2021},
  organization={Springer}
}

Contact

Please, feel free to contact the authors for any unexpected behavior you might face (e.g., bugs). Moreover, if you’re interested in using our code, we appreciate if you cite this work in your project.

• Isabela Borlido: [e-mail]
• Felipe C. Belém: [e-mail]
• Zenilton K. G. Do Patrocínio Jr. [e-mail]
• Alexandre X. Falcão: [e-mail]
• Silvio J. F. Guimarães: [e-mail]