medical-decathlon icon indicating copy to clipboard operation
medical-decathlon copied to clipboard

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

pred152_3D. 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