Framework for Medical Image Segmentation with Convolutional Neural Networks and Deep Learning
MIScnn: Medical Image Segmentation with Convolutional Neural Networks
The open-source Python library MIScnn is an intuitive API allowing fast setup of medical image segmentation pipelines with state-of-the-art convolutional neural network and deep learning models in just a few lines of code.
MIScnn provides several core features:
- 2D/3D medical image segmentation for binary and multi-class problems
- Data I/O, preprocessing and data augmentation for biomedical images
- Patch-wise and full image analysis
- State-of-the-art deep learning model and metric library
- Intuitive and fast model utilization (training, prediction)
- Multiple automatic evaluation techniques (e.g. cross-validation)
- Custom model, data I/O, pre-/postprocessing and metric support
- Based on Keras with Tensorflow as backend
Getting started: 30 seconds to a MIS pipeline
Create a Data I/O instance with an already provided interface for your specific data format.
from miscnn.data_loading.data_io import Data_IO from miscnn.data_loading.interfaces.nifti_io import NIFTI_interface # Create an interface for kidney tumor CT scans in NIfTI format interface = NIFTI_interface(pattern="case_0000[0-2]", channels=1, classes=3) # Initialize data path and create the Data I/O instance data_path = "/home/mudomini/projects/KITS_challenge2019/kits19/data.original/" data_io = Data_IO(interface, data_path)
Create a Preprocessor instance to configure how to preprocess the data into batches.
from miscnn.processing.preprocessor import Preprocessor pp = Preprocessor(data_io, batch_size=4, analysis="patchwise-crop", patch_shape=(128,128,128))
Create a deep learning neural network model with a standard U-Net architecture.
from miscnn.neural_network.model import Neural_Network from miscnn.neural_network.architecture.unet.standard import Architecture unet_standard = Architecture() model = Neural_Network(preprocessor=pp, architecture=unet_standard)
Congratulations to your ready-to-use Medical Image Segmentation pipeline including data I/O, preprocessing and data augmentation with default setting.
Let's run a model training on our data set. Afterwards, predict the segmentation of a sample using the fitted model.
# Training the model with all except one sample for 500 epochs sample_list = data_io.get_indiceslist() model.train(sample_list[0:-1], epochs=500) # Predict the one remaining sample pred = model.predict([sample_list[-1]], direct_output=True)
Now, let's run a 5-fold Cross-Validation with our model, create automatically evaluation figures and save the results into the directory "evaluation_results".
from miscnn.evaluation.cross_validation import cross_validation cross_validation(sample_list, model, k_fold=5, epochs=100, evaluation_path="evaluation_results", draw_figures=True)
There are two ways to install MIScnn:
- Install MIScnn from PyPI (recommended):
Note: These installation steps assume that you are on a Linux or Mac environment. If you are on Windows or in a virtual environment without root, you will need to remove sudo to run the commands below.
sudo pip install miscnn
- Alternatively: install MIScnn from the GitHub source:
First, clone MIScnn using git:
git clone https://github.com/frankkramer-lab/MIScnn.git
Then, cd to the MIScnn folder and run the install command:
cd MIScnn sudo python setup.py install
IT-Infrastructure for Translational Medical Research
How to cite / More information
Dominik Müller and Frank Kramer. (2019)
MIScnn: A Framework for Medical Image Segmentation with Convolutional Neural Networks and Deep Learning.
This project is licensed under the GNU GENERAL PUBLIC LICENSE Version 3.
See the LICENSE.md file for license rights and limitations.
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
|Filename, size||File type||Python version||Upload date||Hashes|
|Filename, size miscnn_TF_1.14-0.32-py3-none-any.whl (94.5 kB)||File type Wheel||Python version py3||Upload date||Hashes View|
|Filename, size miscnn_TF-1.14-0.32.tar.gz (43.0 kB)||File type Source||Python version None||Upload date||Hashes View|
Hashes for miscnn_TF_1.14-0.32-py3-none-any.whl