Med-Imagetools: Transparent and Reproducible Medical Image Processing Pipelines in Python

Med-ImageTools core features

• AutoPipeline CLI
• nnunet nnU-Net compatibility mode
• Built-in train/test split for both normal/nnU-Net modes
• random_state for reproducible seeds
• Region of interest (ROI) yaml dictionary intake for RTSTRUCT processing
• Markdown report output post-processing
• continue_processing flag to continue autopipeline
• dry_run flag to only crawl the dataset

Med-Imagetools, a python package offers the perfect tool to transform messy medical dataset folders to deep learning ready format in few lines of code. It not only processes DICOMs consisting of different modalities (like CT, PET, RTDOSE and RTSTRUCTS), it also transforms them into deep learning ready subject based format taking the dependencies of these modalities into consideration.

Introduction

A medical dataset, typically contains multiple different types of scans for a single patient in a single study. As seen in the figure below, the different scans containing DICOM of different modalities are interdependent on each other. For making effective machine learning models, one ought to take different modalities into account.

Fig.1 - Different network topology for different studies of different patients

Med-Imagetools is a unique tool, which focuses on subject based Machine learning. It crawls the dataset and makes a network by connecting different modalities present in the dataset. Based on the user defined modalities, med-imagetools, queries the graph and process the queried raw DICOMS. The processed DICOMS are saved as nrrds, which med-imagetools converts to torchio subject dataset and eventually torch dataloader for ML pipeline.

Fig.2 - Med-Imagetools AutoPipeline diagram

Installing med-imagetools

pip install med-imagetools

(recommended) Create new conda virtual environment

conda create -n mit
conda activate mit
pip install med-imagetools

(optional) Install in development mode

conda create -n mit
conda activate mit
pip install -e git+https://github.com/bhklab/med-imagetools.git

This will install the package in editable mode, so that the installed package will update when the code is changed.