nnunetv2
nnU-Net is a framework for out-of-the box image segmentation.
Decision gist · record as of 2026-08-14
Yes. nnU-Net is actively maintained, has low install friction, carries a permissive Apache 2.0 license, and reports no known vulnerabilities. It is well-suited for anyone working on biomedical image segmentation who wants to avoid manual architecture tuning. The substantial dependency footprint (22 runtime packages) is appropriate for its domain and is unlikely to conflict with typical scientific Python workflows. Start with the installation and getting-started documentation.AI-flagged interpretation of the facts on this page — verify before relying
Before you install
- Requires PyTorch to be installed first for your hardware (CPU/GPU).
- Requires Python >= 3.10.
- Needs environment setup for nnUNet_raw, nnUNet_preprocessed, and nnUNet_results directories per documentation.
License · maintenance · safety
permissive license (permissive) — Apache License 2.0 is permissive: you can use, modify, and distribute nnU-Net freely in commercial and private projects, provided you include the license and attribute the original copyright holders.
last release 2026-07-01 (44 days) · last repo commit 2026-07-23 · 8,788 stars
0 known vulnerabilities (OSV.dev, 2026-08-14) · 155,255 downloads/mo, #10,827 on PyPI
Alternatives
Verify before relying
pip install nnunetv2
from nnunetv2.imageio.base_reader_writer import BaseReaderWriter
from nnunetv2.training.nnUNetTrainer.nnUNetTrainer import nnUNetTrainer- Whether the framework supports inference on pre-trained models from external sources or only on models trained within nnU-Net
- Performance characteristics and memory requirements for typical biomedical datasets
- Specific GPU/CUDA version compatibility beyond the Python requirement
What it is and what it does
nnU-Net is a self-configuring deep learning framework for semantic segmentation, primarily designed for biomedical image analysis. It analyzes your training data to create a dataset fingerprint, automatically selects and configures appropriate U-Net architectures (including 2D, 3D, and cascade variants), and handles the full pipeline from preprocessing through training to model selection and inference. The framework is built on PyTorch and depends on a substantial stack of scientific libraries including scipy, scikit-learn, SimpleITK, nibabel, and others for image I/O, processing, and visualization.
It works particularly well in training-from-scratch scenarios on biomedical, medical imaging, and challenge datasets where standard pretrained models are often a poor fit. While primarily designed for supervised segmentation tasks, it serves as both a production tool and a development framework for researchers exploring new segmentation methods. The package is actively maintained by the Applied Computer Vision Lab at Helmholtz Imaging and the Division of Medical Image Computing at the German Cancer Research Center.
Use it for
- Segment organs, tumors, or anatomical structures in CT, MRI, or other medical imaging modalities without manual architecture design
- Establish a strong baseline for new segmentation datasets or medical imaging challenges
- Train and compare multiple U-Net configurations automatically on your own biomedical image data
- Preprocess and standardize medical imaging datasets for downstream analysis or model development
- Perform 3D volumetric segmentation on stacked 2D slices or native 3D imaging data
Worth the install?
AI-flagged interpretation of the facts on this page. Verify before relying on it.
Yes.
nnU-Net is actively maintained, has low install friction, carries a permissive Apache 2.0 license, and reports no known vulnerabilities. It is well-suited for anyone working on biomedical image segmentation who wants to avoid manual architecture tuning. The substantial dependency footprint (22 runtime packages) is appropriate for its domain and is unlikely to conflict with typical scientific Python workflows. Start with the installation and getting-started documentation.
Install
nnunetv2 on PyPI
Before you install
Low friction installation with a pure Python wheel. Active maintenance with a recent release 44 days ago and ongoing repository activity. Requires torch and 22 runtime dependencies including scipy, scikit-learn, and medical imaging libraries like SimpleITK and nibabel.
Requires PyTorch to be installed first for your hardware (CPU/GPU). Requires Python >= 3.10. Needs environment setup for nnUNet_raw, nnUNet_preprocessed, and nnUNet_results directories per documentation.
License in practice
Apache License 2.0 is permissive: you can use, modify, and distribute nnU-Net freely in commercial and private projects, provided you include the license and attribute the original copyright holders.
Quickstart
pip install nnunetv2
from nnunetv2.imageio.base_reader_writer import BaseReaderWriter
from nnunetv2.training.nnUNetTrainer.nnUNetTrainer import nnUNetTrainerVerify before relying
- Whether the framework supports inference on pre-trained models from external sources or only on models trained within nnU-Net
- Performance characteristics and memory requirements for typical biomedical datasets
- Specific GPU/CUDA version compatibility beyond the Python requirement
Package facts
| License | permissive license permissive |
| Python support | Supports the current Python release >=3.10 |
| Install friction | Low. Pure-Python wheel |
| Runtime dependencies | 22 packagestorchacvl-utilsdynamic-network-architecturestqdmscipybatchgeneratorsnumpyscikit-learnscikit-imageSimpleITKpandasgraphviztifffilerequestsnibabelmatplotlibseabornimagecodecsyacsbatchgeneratorsv2einopsblosc2 |
| Maintenance | Actively maintained 44 days since the last release |
| Last repo commit | |
| First released | |
| Downloads | 155,255 / month, #10,827 on PyPI 30-day window, as of 2026-08-14 |
| Known vulnerabilities | None known OSV.dev, checked 2026-08-14 |
| Classifiers | Development Status :: 5 - Production/StableIntended Audience :: DevelopersIntended Audience :: Healthcare IndustryIntended Audience :: Science/ResearchLicense :: OSI Approved :: Apache Software LicenseProgramming Language :: Python :: 3Topic :: Scientific/Engineering :: Artificial IntelligenceTopic :: Scientific/Engineering :: Image RecognitionTopic :: Scientific/Engineering :: Medical Science Apps. |
Evidence: nnunetv2-2.8.1-py3-none-any.whl
Tags
Let your AI agent find packages like this
Example. Real query, live index.
You found this page by searching. An agent finds it by wishing: SkillFed indexes 14,416 PyPI packages by what they can do, searchable in plain language.
wish › “semantic segmentation framework”
- nnunetv2nnU-Net is a semantic segmentation framework that automatically…
- segmentation-models-pytorchProvides PyTorch-based neural network models for image semantic…
- ultralyticsUltralytics YOLO provides a unified framework for training and…
Give your agent the search over MCP, or paste the wish link into any chat.
More Artificial Intelligence packages
LiteLLM provides a unified Python interface to call 100+ LLM providers (OpenAI, Anthropic, Gemini, Bedrock, Azure, and others) using OpenAI-compatible API format, available as both a Python SDK and a self-hosted AI Gateway proxy server.
Install it if you need to work with multiple LLM providers or want to centralize LLM routing in your organization.
Client library and CLI tool for downloading, uploading, and managing models, datasets, and repositories on the Hugging Face Hub platform.
Install it if you work with Hugging Face Hub models or datasets.
LangChain provides a framework for building agents and LLM-powered applications by composing language models, tools, and memory through a unified API that abstracts over multiple model providers.
hf-xet provides chunk-based deduplication and efficient file transfer for the Hugging Face Hub, enabling faster uploads and downloads of large files with local disk caching.
Tokenizers converts raw text into token sequences for NLP models, with support for training custom vocabularies and using pre-built tokenizers (BPE, WordPiece) optimized for speed via Rust.
Transformers provides a unified framework for loading, fine-tuning, and running state-of-the-art pretrained models across text, vision, audio, video, and multimodal tasks using PyTorch, JAX, or TensorFlow.
Install it if you need to run or train any transformer-based model for NLP, vision, audio, or multimodal tasks.
See also segmentation-models-pytorch · TotalSegmentator · pytorchcv · cellpose · connected-components-3d · monai · ultralytics · batchgeneratorsv2 · nuscenes-devkit · itk-segmentation