SOTA 3D Medical Segmentation framework (UNET++, TransUNET , SwimTransUNET) in TensorFlow. Fully TPU-optimized with a 3-stage OHEM pipeline for the Medical Segmentation Decathlon (MSD).
computer-vision deep-learning tensorflow keras medical-imaging msd 3d-segmentation tpu image-segmentation-tensorflow ohem mlops brain-tumour swin-transformer transunet medical-segmentation-decathlon unet-plus-plus
-
Updated
Dec 8, 2025 - Python