intel / unet

U-Net Biomedical Image Segmentation
Apache License 2.0
303 stars 124 forks source link
artificial-intelligence-algorithms deep-learning keras tensorflow u-net

DISCONTINUATION OF PROJECT

This project will no longer be maintained by Intel.
Intel has ceased development and contributions including, but not limited to, maintenance, bug fixes, new releases, or updates, to this project.
Intel no longer accepts patches to this project.
If you have an ongoing need to use this project, are interested in independently developing it, or would like to maintain patches for the open source software community, please create your own fork of this project.

Deep Learning Medical Decathlon Demos for Python*

U-Net Biomedical Image Segmentation with Medical Decathlon Dataset.

This repository contains 2D and 3D U-Net TensorFlow scripts for training models using the Medical Decathlon dataset (http://medicaldecathlon.com/).

![pred152_3D](3D/images/BRATS_152_img3D.gif "BRATS image #152: Purple voxels indicate a perfect prediction by the model. Red are false positives. Blue are false negatives"). pred195

Citation

David Ojika, Bhavesh Patel, G. Athony Reina, Trent Boyer, Chad Martin and Prashant Shah. “Addressing the Memory Bottleneck in AI Model Training”, Workshop on MLOps Systems, Austin TX (2020) held in conjunction with Third Conference on Machine Learning and Systems (MLSys). https://arxiv.org/abs/2003.08732