Fast and lightweight TotalSegmentator CLI powered by the KonfAI framework.
Project description
TotalSegmentator-KonfAI
Fast and lightweight TotalSegmentator inference using the KonfAI framework
🧩 Overview
TotalSegmentator-KonfAI is a lightweight command-line interface (CLI) for running TotalSegmentator models a multi-organ medical image segmentation, through the KonfAI deep learning framework.
It provides fast inference for segmentation tasks, even on low-resource hardware.
Models are automatically downloaded from Hugging Face Hub.
🚀 Installation
From PyPI:
pip install totalsegmentator-konfai
From source:
git clone https://github.com/vboussot/TotalSegmentator-KonfAI.git
cd TotalSegmentator-KonfAI
pip install .
⚙️ Usage
Perform segmentation on an input volume:
totalsegmentator-konfai -i path/to/image.nii.gz -o path/to/seg.nii.gz
Optional arguments
| Flag | Description | Default |
|---|---|---|
-i, --input |
Path to the input medical image | required |
-o, --output |
Path to save the segmentation | Seg.nii.gz |
-ta, --task |
Choose model type: total or total_mr |
total |
-f, --fast |
Use faster low-resolution model (≈3 mm) | False |
-g, --gpu |
GPU list (e.g. 0 or 0,1) |
CPU if unset |
--cpu |
Number of CPU cores (used if no GPU) | 1 |
-q, --quiet |
Suppress console output | False |
Example
totalsegmentator-konfai -i patient01.nii.gz -o seg_patient01.nii.gz --gpu 0 --task total --fast
🧠 Features
- ⚡ Fast inference using KonfAI
- 🤗 Automatic model download from Hugging Face
- 🧩 Multi-model support
- 🧾 Multi-format compatibility: supports all major medical image formats handled by ITK
📖 Reference
This package is based on the original TotalSegmentator by
Wasserthal et al., a deep learning framework for whole-body CT and MR segmentation.
For scientific use, please cite the original TotalSegmentator work in addition to this CLI tool.
🧾 License
Released under the Apache 2.0 License.
The original TotalSegmentator and KonfAI licenses remain with their respective authors.
🔗 Links
- 🧠 Original TotalSegmentator: github.com/wasserth/TotalSegmentator
- ⚙️ KonfAI Framework: github.com/vboussot/KonfAI
- 🤗 Model Hub: huggingface.co/VBoussot/TotalSegmentator-KonfAI
- 📦 PyPI Package: pypi.org/project/totalsegmentator-konfai
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file totalsegmentator_konfai-1.4.5.tar.gz.
File metadata
- Download URL: totalsegmentator_konfai-1.4.5.tar.gz
- Upload date:
- Size: 11.9 kB
- Tags: Source
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/6.1.0 CPython/3.13.7
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
58bbd1e4bd4dcbc4c0b1945755069004792c68a12e1506b0f1da1aa5dff49286
|
|
| MD5 |
5829fdb3d356201c295a80503a577b5f
|
|
| BLAKE2b-256 |
8c030321b3afc8a6f6a2724491eb71fb56ddb5a78b814969063f7b128cfa7778
|
Provenance
The following attestation bundles were made for totalsegmentator_konfai-1.4.5.tar.gz:
Publisher:
publish.yml on vboussot/KonfAI
-
Statement:
-
Statement type:
https://in-toto.io/Statement/v1 -
Predicate type:
https://docs.pypi.org/attestations/publish/v1 -
Subject name:
totalsegmentator_konfai-1.4.5.tar.gz -
Subject digest:
58bbd1e4bd4dcbc4c0b1945755069004792c68a12e1506b0f1da1aa5dff49286 - Sigstore transparency entry: 785146052
- Sigstore integration time:
-
Permalink:
vboussot/KonfAI@6c5762ac1f22d7abee44f2a6f48241bc15d1863a -
Branch / Tag:
refs/tags/v1.4.5 - Owner: https://github.com/vboussot
-
Access:
public
-
Token Issuer:
https://token.actions.githubusercontent.com -
Runner Environment:
github-hosted -
Publication workflow:
publish.yml@6c5762ac1f22d7abee44f2a6f48241bc15d1863a -
Trigger Event:
push
-
Statement type:
File details
Details for the file totalsegmentator_konfai-1.4.5-py3-none-any.whl.
File metadata
- Download URL: totalsegmentator_konfai-1.4.5-py3-none-any.whl
- Upload date:
- Size: 11.8 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/6.1.0 CPython/3.13.7
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
1915f128763ffd7a6ff5aa58645ae1e21f81ac5ed430b236d481c95b450a6802
|
|
| MD5 |
51e2d7fd6fae62686b5ed2f08082261c
|
|
| BLAKE2b-256 |
679b62de3c08e15d6201ca95de5fee9bf66050f3cc3b00a676885ab0244c7eeb
|
Provenance
The following attestation bundles were made for totalsegmentator_konfai-1.4.5-py3-none-any.whl:
Publisher:
publish.yml on vboussot/KonfAI
-
Statement:
-
Statement type:
https://in-toto.io/Statement/v1 -
Predicate type:
https://docs.pypi.org/attestations/publish/v1 -
Subject name:
totalsegmentator_konfai-1.4.5-py3-none-any.whl -
Subject digest:
1915f128763ffd7a6ff5aa58645ae1e21f81ac5ed430b236d481c95b450a6802 - Sigstore transparency entry: 785146068
- Sigstore integration time:
-
Permalink:
vboussot/KonfAI@6c5762ac1f22d7abee44f2a6f48241bc15d1863a -
Branch / Tag:
refs/tags/v1.4.5 - Owner: https://github.com/vboussot
-
Access:
public
-
Token Issuer:
https://token.actions.githubusercontent.com -
Runner Environment:
github-hosted -
Publication workflow:
publish.yml@6c5762ac1f22d7abee44f2a6f48241bc15d1863a -
Trigger Event:
push
-
Statement type: