Skip to content

Segmantic

Segmantic is a MONAI/PyTorch-based library for medical image segmentation.

Features

  • utilities to prepare datasets for use in training of segmentation and style transfer networks
  • segmentation networks
  • evaluation metrics (fast confusion matrix, Hausdorff distance, ...)
  • image to image translation (based on pix2pix / cyclegan) to help generalization from limited training data

Installation

Create a virtual environment

cd /your/working/directory
python -m venv .venv

Activate it using e.g. source .venv/bin/activate on Linux/Mac and .venv\Scripts\activate.bat on Windows.

To install this repo (this will install all dependencies):

pip install git+https://github.com/dyollb/segmantic.git#egg=segmantic

Or in edit/dev mode

pip install -e git+https://github.com/dyollb/segmantic.git#egg=segmantic[dev]

Getting started

The project layout is as follows:

|-- segmantic/src/segmantic
    |-- commands   # command lines
    |-- i2i        # image-to-image translation (style transfer)
    |-- prepro     # module containing utils to prepare your data
    |-- seg        # semantic segmentation: training, inference and evaluation
    |-- util       # utility functions
|-- segmantic/pyproject.toml
|-- segmantic/scripts
|-- segmantic/tests