Metadata-Version: 2.4
Name: magdi-segmentation-models-3d
Version: 0.0.0
Summary: MAGDI Segmentation Models 3D
Author-email: Christian Hänig <christian.haenig@hs-anhalt.de>, Christian Gurski <christian.gurski@hs-anhalt.de>
Project-URL: Gitlab, https://gitlab.hs-anhalt.de/ki/projekte/magdi/magdi-data
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.12
Classifier: Operating System :: OS Independent
Requires-Python: >=3.12
Description-Content-Type: text/markdown; charset=UTF-8
Requires-Dist: torch>=2.6.0
Requires-Dist: torchvision>=0.21.0
Requires-Dist: numpy>=1.26.3
Requires-Dist: monai>=1.4.0
Requires-Dist: einops>=0.8.1
Requires-Dist: python-dotenv>=1.0.1
Requires-Dist: typing_extensions>=4.13.2
Requires-Dist: pyyaml>=6.0.2
Requires-Dist: transformers>=4.55.2
Requires-Dist: huggingface-hub>=0.34.4
Requires-Dist: pillow>=11.2.1
Requires-Dist: optree>=0.17.0

# MAGDI Segmentation Models 3D
This python package named ``magdi_segmentation_models_3d`` responsible for providing
custom Hugging Face compatible models for 3D image segmentation.

## Hugging Face models

Documentation on Hugging Face: https://huggingface.co/docs/transformers/en/custom_models

Examples:
https://github.com/huggingface/transformers/tree/main/src/transformers/models

### swinunetrv2
Our swinunetrv2 implementation based on monai SwinUNETR

### unet
Our unet implementation based on monai UNet
