This release is a pre-release and may not be stable for production use.
pyplatypus
Computer vision for medical imaging — the engine behind the platypus R package.
0.3.0a1 — an alpha. This replaces the 2022 TensorFlow package with a PyTorch one. The API will still move and the R surface does not exist yet, so pin the exact version if you build on it.
Everything on PyPI so far is a pre-release, so
pip install pyplatypusresolves to this one.pip install pyplatypus==0.1.0rc2gets the old TensorFlow package.
What works today
Semantic segmentation in 2D, end to end:
- One spec, two ways in. Build it from arguments or load it from YAML — both produce the same object, so nothing downstream can tell which you used.
- Four architectures: U-Net, U-Net++, Res-U-Net, LinkNet, each composable with separable convolutions, spatial dropout, learned or interpolated upsampling, deep supervision and configurable block width.
- Nine losses (IoU, Dice, CCE, CCE-Dice, Focal, Tversky, Focal-Tversky, Combo, Lovász) and three metrics, all reducing over every axis except batch and channel — so they already work on volumes.
- Tiling that goes both ways: cut a large image into a grid instead of shrinking it, and get a full-size mask back.
- DICOM, read properly: stored values converted to real units through the modality LUT, a fixed window rather than one taken from the image, and MONOCHROME1 inverted. Opening the file is the easy part; those three are what make two scans comparable.
- Splitting by patient, not by slice:
split_dataset()turns one directory into train/validation/test CSVs, keeping every group whole. Slices of one patient on both sides of a split is the most common way to publish a Dice that means nothing, and agroup_bythat matches nothing is an error rather than a silent fallback. - Scores per case, not one number:
evaluate_cases()reports each image or patient separately andsummarise_cases()gives the distribution and names the worst one. On the Data Science Bowl a model at 0.855 over the split turned out to score 0.006 on three images; the mean had no way of saying so. - Volumes, not only slices: NIfTI in, 3D U-Net out, patches through the same tiling
that cuts 2D images. Every volume is reoriented to canonical (RAS) first - a NIfTI's
affine, not its array order, says where the anatomy is, and two datasets read naively can
be mirror images of each other. Masks may be label maps (
labels: [0, 1]) as well as pictures, because that is how every volume format stores them. - Many models from one file, with a comparison table at the end.
Object detection, ensembling and pretrained backbones remain out of scope. So does augmentation in 3D: albumentations ships 3D transforms through a different call signature, and a 3D spec asking for augmentation is refused rather than silently ignored.
Also not done yet, deliberately: resampling to isotropic spacing. Spacing is read and
carried (volume_spacing()), because losing it would make resampling impossible later, but
applying it changes the voxel grid the model sees and that is a decision to make explicitly
rather than inside a reader.
Try it
uv venv --python 3.11 .venv
uv pip install --python .venv/bin/python -e ".[dev]"
.venv/bin/python -m pytest
# The linter is pinned in pyproject.toml and installed on its own, so it agrees with CI
# and does not drag torch along to read text files.
uv run --only-group lint ruff check pyplatypus tests
from pyplatypus import Engine, from_yaml
engine = Engine(from_yaml("examples/data_science_bowl.yaml"))
engine.fit(verbose=True)
for row in engine.evaluate():
print(row)
masks = engine.predict(engine.best_model("dice"), split="test")
examples/data_science_bowl.yaml trains a U-Net and a LinkNet on the 2018 Data Science
Bowl and prints a comparison. On a GTX 1070 that is about 11 seconds per epoch at
160×160.
Requirements
Python ≥ 3.10, and torch ≥ 2.7.
If your GPU is a GTX 10-series (Pascal) or older, install the pascal extra:
pip install "pyplatypus[pascal]"
torch 2.8 and later ship CUDA 13 builds, and CUDA 13 dropped the Maxwell, Pascal and Volta generations outright - no driver update brings them back. The last torch built against CUDA 12 is 2.7.x, which the extra pins. On anything from Turing (RTX 20-series) onwards, ignore this.
Licence
MIT.
1 maintainer
Metadata
1 maintainer
Release files for pyplatypus 0.3.0a1
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | Provenance | |
|---|---|---|---|---|
| pyplatypus-0.3.0a1.tar.gz | 92.5 kB | With provenance | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| pyplatypus-0.3.0a1-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 157.5 kB
Release files / pyplatypus-0.3.0a1.tar.gz
| Download URL | pyplatypus-0.3.0a1.tar.gz |
|---|---|
| Size | 92.5 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
0bdb61327e9ea8a014e619eb549af39acb6c2a991c750cb1c717e25076ff3504
|
|
BLAKE2b-256 checksum How to use checksums |
9df98acbf7ddb13c9b6750cefea65a6a7095784b3beb251124ecac2cc2bc4490
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
twine/7.0.0 CPython/3.13.14
|
Provenance
Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.
PyPI Publish Attestation
PyPI verified that this artifact, at this checksum, originated from the publisher listed below.
Signed by GitHub Actions, verified by PyPI on Sep 26, 2026.
Transparency logRelease files / pyplatypus-0.3.0a1-py3-none-any.whl
| Download URL | pyplatypus-0.3.0a1-py3-none-any.whl |
|---|---|
| Size | 65.1 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
c345c4d10088d5e1e0e5ac35258f14c85c1ecbfcdafdef73a46f3f066c772168
|
|
BLAKE2b-256 checksum How to use checksums |
368fa40ebddef2fa0cd2c09ea44044f2628571bb577725ba670bc5a3c3dab023
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
twine/7.0.0 CPython/3.13.14
|
Provenance
Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.
PyPI Publish Attestation
PyPI verified that this artifact, at this checksum, originated from the publisher listed below.
Signed by GitHub Actions, verified by PyPI on Sep 26, 2026.
Transparency logSecurity for pyplatypus 0.3.0a1
Provenance
Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.
Full provenance
All 2 files in this release have been uploaded with provenance metadata.
Attestation for pyplatypus-0.3.0a1.tar.gz
PyPI Publish Attestation
PyPI verified that this artifact, at this checksum, originated from the publisher listed below.
Signed by GitHub Actions, verified by PyPI on Sep 26, 2026.
Transparency logA note on package security
PyPI is a repository for distributing packages, not a service for auditing them. While third-party security researchers and our staff do their best to detect and remove malicious packages, PyPI cannot guarantee the quality, security or suitability of any given package.
To learn more about protecting yourself from malicious actors, visit our package security documentation.
