oak2

oak2 is a cross-platform C++ implementation of Random Forests supporting local and global image classification and regression. Local prediction models can be employed for problems such as image segmentation and image synthesis, while global models enable image-level predictions.

Dependencies

oak2 depends on several third-party libraries:

• Eigen
• Intel TBB (tested up to v.4.4)
• Boost (tested up to v1.58)
• ITK (tested up to v4.13.2)

Build instructions

Eigen is a header-only library and can be simply installed via:

#!bash

$ mkdir 3rdparty
$ cd 3rdparty
$ wget https://gitlab.com/libeigen/eigen/-/archive/3.3.7/eigen-3.3.7.tar.gz --progress=bar:force:noscroll
$ mkdir eigen
$ tar xf eigen-3.3.7.tar.gz -C eigen --strip-components=1

You can download and install ITK in the same 3rdparty folder via:

#!bash

$ wget https://sourceforge.net/projects/itk/files/itk/5.0/InsightToolkit-5.0.0.tar.gz
$ tar xvf InsightToolkit-5.0.0.tar.gz
$ cd InsightToolkit-5.0.0
$ mkdir build
$ cd build
$ cmake -DCMAKE_INSTALL_PREFIX=../../itk ..
$ make -j4
$ make install

Alternatively, you can check out these ITK install instructions.

You can install Boost and TBB via apt-get:

#!bash

$ sudo apt-get install libboost-all-dev libtbb-dev

Note, you might have to specify a specific version via apt-get install <package>=<version>.

oak2 comes with a CMake configuration file. From the top folder where CMakeLists.txt is located (same as this README), do the following to build all internal libraries and executables:

#!bash

$ mkdir build
$ cd build
$ export THIRD_PARTY_DIR=<folder_containing_eigen_and_itk>
$ cmake ..
$ make -j4

The command line tools are located in build/oak/apps.

Build with Docker

To build the docker image run

docker build -t oak2:latest .

Then run it with docker by mounting your local volume <volume> and calling an executable <exec> with its associated input arguments:

docker run -it --rm -v <volume>:/workdir oak2:latest <exec> --help

<exec> can be one of the following:

# local regression
oak2_local_reg_trainer
oak2_local_reg_tester

# local classification
oak2_local_class_trainer
oak2_local_class_tester

# global regression
oak2_global_reg_trainer
oak2_global_reg_tester

# global classification
oak2_global_class_tester
oak2_global_class_trainer

For example, to call oak2_global_class_trainer with the local volume mount /media/SSD/data, run

docker run -it --rm -v /media/SSD/data:/workdir oak2:latest oak2_global_class_trainer --help

Usage

Detailed instructions and examples are coming soon...

Acknowledgements

Special thanks go to Hauke Heibel who has contributed significantly to the C++ implementation.