Skip to main content

Fast and lightweight TotalSegmentator CLI powered by the KonfAI framework.

Project description

PyPI version Python License KonfAI

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


Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

totalsegmentator_konfai-1.4.5.tar.gz (11.9 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

totalsegmentator_konfai-1.4.5-py3-none-any.whl (11.8 kB view details)

Uploaded Python 3

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

Hashes for totalsegmentator_konfai-1.4.5.tar.gz
Algorithm Hash digest
SHA256 58bbd1e4bd4dcbc4c0b1945755069004792c68a12e1506b0f1da1aa5dff49286
MD5 5829fdb3d356201c295a80503a577b5f
BLAKE2b-256 8c030321b3afc8a6f6a2724491eb71fb56ddb5a78b814969063f7b128cfa7778

See more details on using hashes here.

Provenance

The following attestation bundles were made for totalsegmentator_konfai-1.4.5.tar.gz:

Publisher: publish.yml on vboussot/KonfAI

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file totalsegmentator_konfai-1.4.5-py3-none-any.whl.

File metadata

File hashes

Hashes for totalsegmentator_konfai-1.4.5-py3-none-any.whl
Algorithm Hash digest
SHA256 1915f128763ffd7a6ff5aa58645ae1e21f81ac5ed430b236d481c95b450a6802
MD5 51e2d7fd6fae62686b5ed2f08082261c
BLAKE2b-256 679b62de3c08e15d6201ca95de5fee9bf66050f3cc3b00a676885ab0244c7eeb

See more details on using hashes here.

Provenance

The following attestation bundles were made for totalsegmentator_konfai-1.4.5-py3-none-any.whl:

Publisher: publish.yml on vboussot/KonfAI

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page