Image Segmentation
medical
biology
vascx / README.md
josedvq's picture
Update README.md
5521c8a verified
metadata
license: agpl-3.0
pipeline_tag: image-segmentation
tags:
  - medical
  - biology

VascX models

This repository contains the instructions for using the VascX models from the paper VascX Models: Model Ensembles for Retinal Vascular Analysis from Color Fundus Images.

The model weights are in huggingface.

Installation

To install the entire fundus analysis pipeline including fundus preprocessing, model inference code and vascular biomarker extraction:

  1. Create a conda or virtualenv virtual environment, or otherwise ensure a clean environment.

  2. Install the rtnls_inference package.

Usage

To speed up re-execution of vascx we recommend to run the preprocessing and segmentation steps separately:

  1. Preprocessing. See this notebook. This step is CPU-heavy and benefits from parallelization (see notebook).

  2. Inference. See this notebook. All models can be ran in a single GPU with >10GB VRAM.