IntelAI / unet

U-Net Biomedical Image Segmentation

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/).

.

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