Prostate Segmentation MoE

A simple Python module for easy segmentation of the prostate in T2-weighted MRI images in NIfTI format. This module utilizes a mixture of U-Net architectures for segmentation tasks and aims to provide a straightforward solution for users working with prostate MRI data.

Features

• Segmentation of prostate regions in T2-weighted MRI images.
• Uses a combination of trained U-Net models for accurate and efficient segmentation.
• Supports input in the NIfTI format, a common format for medical imaging.
• Returns 3D masks aswell as the volume estimation.

Recommended before installation (Windows)

Install virtualenv if not already

pip install virtualenv

Create a virtual environment

python -m venv venv

Activate it

venv\Scripts\activate

• You may need to run Set-ExecutionPolicy Unrestricted -Scope Process before activation
• If creating your own repository don't forget to ignore the venv folder in the .gitignore file

Installation

Install the library using pip:

pip install psmoe

Usage

Example

Check for example.py

Scripted Download

You can also download a T2-weighted MRI sequence using the provided function in the example.

License

This project is licensed under the MIT License - see the LICENSE file for details.

For more details, visit the GitHub repository.