Deep learning based, intraretinal fluid segmenter for macular OCTs
Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Failed to load latest commit information.
Dockerfile
LICENSE
README.md
build-docker.sh
example.png
output-blend.png
output-mask.png
output-mask_blend.png
params.txt
requirements.txt
run-docker.sh
run.py
weights.hdf5

README.md

irf-segmenter

License: GPL v3

Fully-automated, deep learning based, intraretinal fluid segmenter for optical coherence (OCT) images of the macula

Usage

usage: run.py [-h] [--mode MODE] input_file output_file

positional arguments:
  input_file   Input PNG file
  output_file  Output PNG file

optional arguments:
  -h, --help   show this help message and exit
  --mode MODE  Output mode, 'mask_blend' (default) for masked heatmap output,
               'mask' for binary mask output, 'blend' for blended heatmap

Example

Input image:

input image example

Mask Blended image:

input image example

Blended image:

input image example

Binary Mask image:

input image example

Requirements

You must install the following Python packages:

conda install --file requirements.txt

or

pip install -r requirements.txt

If having troubles with import cv2, try:

conda install --channel https://conda.anaconda.org/menpo opencv3

Installation

git clone https://github.com/uw-biomedical-ml/irf-segmenter