This repository contains two NVIDIA medical segmentation foundation models for 3D medical image segmentation.
Both models follow the MONAI bundle architecture.
| Feature | NV-Segment-CT | NV-Segment-CTMR |
|---|---|---|
| Anatomical Classes | 132 classes | 345+ classes |
| Modalities | CT only | CT + MRI (body & brain) |
| Segmentation Type | Automatic + Interactive (point-click) | Automatic only |
| Model Weights | NV-Segment-CT on HuggingFace | NV-Segment-CTMR on HuggingFace |
NV-Segment-CT (Paper) is a foundation model trained systematically on 11,454 volumes encompassing 127 types of human anatomical structures and various lesions. The model provides State-of-the-art performances on:
- out-of-the-box automatic segmentation on 3D CT scans
- zero-shot interactive segmentation in 3D CT scans
- automatic segemntation + interactive refinement
NV-Segment-CTMR starts from NV-Segment-CT checkpoint and finetuned on over 30K CT and MRI scans, supporting over 300 classes.
- out-of-the-box automatic segmentation on 3D CT scans
- share the same architecture with VISTA3D-CT model but we only trained the automatic segmentation branch with larger CT and MRI datasets.
- Quick Deployment: NVIDIA NIM for NV-Segment-CT - Managed API endpoint
- Documentation: See individual model folders for detailed docs
- Research Paper: VISTA3D: Versatile Imaging SegmenTation and Annotation
- Built with: MONAI - Medical Open Network for AI
