medaugmentx 0.2.0__tar.gz

medaugmentx-0.2.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 MedAugment Contributors
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

medaugmentx-0.2.0/PKG-INFO

@@ -0,0 +1,330 @@
+Metadata-Version: 2.4
+Name: medaugmentx
+Version: 0.2.0
+Summary: Clinically-aware medical image augmentation for AI training (MRI, CT, X-Ray, Tomosynthesis).
+Author: MedAugmentX Contributors
+License: MIT
+Project-URL: Homepage, https://github.com/medaugmentx/medaugmentx
+Project-URL: Documentation, https://github.com/medaugmentx/medaugmentx/tree/main/docs
+Project-URL: Source, https://github.com/medaugmentx/medaugmentx
+Project-URL: Issues, https://github.com/medaugmentx/medaugmentx/issues
+Keywords: medical-imaging,data-augmentation,deep-learning,dicom,nifti,tomosynthesis,mri,ct,radiology-ai
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: Science/Research
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
+Classifier: Topic :: Scientific/Engineering :: Image Processing
+Classifier: Topic :: Scientific/Engineering :: Medical Science Apps.
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: numpy>=1.22
+Requires-Dist: scipy>=1.9
+Provides-Extra: dicom
+Requires-Dist: pydicom>=2.3; extra == "dicom"
+Provides-Extra: nifti
+Requires-Dist: nibabel>=4.0; extra == "nifti"
+Provides-Extra: io
+Requires-Dist: pydicom>=2.3; extra == "io"
+Requires-Dist: nibabel>=4.0; extra == "io"
+Provides-Extra: test
+Requires-Dist: pytest>=7.0; extra == "test"
+Requires-Dist: pytest-cov>=4.0; extra == "test"
+Requires-Dist: hypothesis>=6.70; extra == "test"
+Requires-Dist: pydicom>=2.3; extra == "test"
+Requires-Dist: nibabel>=4.0; extra == "test"
+Provides-Extra: yaml
+Requires-Dist: pyyaml>=6.0; extra == "yaml"
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0; extra == "dev"
+Requires-Dist: pytest-cov>=4.0; extra == "dev"
+Requires-Dist: hypothesis>=6.70; extra == "dev"
+Requires-Dist: pydicom>=2.3; extra == "dev"
+Requires-Dist: nibabel>=4.0; extra == "dev"
+Requires-Dist: pyyaml>=6.0; extra == "dev"
+Requires-Dist: ruff>=0.4; extra == "dev"
+Requires-Dist: mypy>=1.5; extra == "dev"
+Dynamic: license-file
+
+# MedAugmentX
+
+**Clinically-aware medical image augmentation for AI training.**
+
+[![Python](https://img.shields.io/badge/python-3.9%20%E2%80%93%203.14-blue.svg)](https://www.python.org/downloads/)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](LICENSE)
+[![Status: Phase 2](https://img.shields.io/badge/status-phase%202-green.svg)](docs/MILESTONES.md)
+
+MedAugmentX is a purpose-built Python library for data augmentation of medical
+images. Unlike general-purpose libraries (`torchvision`, `albumentations`), it
+treats the unique properties of medical data — anisotropic 3D volumes,
+modality-specific artifacts, mask consistency, and clinical I/O — as
+first-class concerns.
+
+**Zero deep-learning dependencies.** The core library requires only NumPy and
+SciPy. PyTorch, MONAI, and TorchIO interop are planned for Phase 3.
+
+---
+
+## Why MedAugmentX
+
+| Problem with general-purpose augmentation | What MedAugmentX does |
+| --- | --- |
+| Treats every image as 2D RGB | Native 3D volumes with anisotropic-aware ops |
+| No MRI-specific noise or artifact models | Physics-based Rician noise, bias field, k-space corruption, ghosting |
+| CT/DXR augmentation is generic brightness shift | Window/level variation, beam hardening, resolution simulation |
+| Masks drift on complex transforms | Deterministic seeding; image and mask share the same random draw |
+| No tomosynthesis support | Dedicated DBT module with slab-aware augmentations |
+| DICOM/NIfTI I/O scattered across libraries | One loader, one `MedVolume`, vendor metadata preserved |
+| Pipelines can't be saved or reloaded | Full JSON/YAML serialisation with round-trip reconstruction |
+
+---
+
+## Install
+
+```bash
+# Core (NumPy + SciPy only)
+pip install medaugmentx
+
+# With DICOM/NIfTI I/O
+pip install "medaugmentx[io]"
+
+# With YAML serialisation support
+pip install "medaugmentx[yaml]"
+```
+
+Verify the installation:
+
+```python
+import medaugmentx
+print(medaugmentx.__version__)   # 0.2.0
+```
+
+---
+
+## Quick start
+
+### One-line preset
+
+The fastest way to get started is a pre-built modality preset:
+
+```python
+from medaugmentx.presets import mri_pipeline, ct_pipeline, dxr_pipeline, dbt_pipeline
+
+pipeline = mri_pipeline(seed=42)
+augmented = pipeline(vol)          # MedVolume in, MedVolume out
+```
+
+Four presets ship out of the box — see [Preset pipelines](#preset-pipelines) below.
+
+### Build your own pipeline
+
+```python
+import numpy as np
+from medaugmentx import MedVolume, Compose, OneOf
+from medaugmentx.transforms import (
+    RandomAffine, ElasticDeform,
+    BiasField, RicianNoise, GaussianNoise, GammaCorrection,
+)
+
+vol = MedVolume(
+    image=np.random.rand(80, 256, 256).astype(np.float32),
+    mask=np.zeros((80, 256, 256), dtype=np.uint8),
+    spacing=(1.0, 0.7, 0.7),          # mm per axis (z, y, x)
+    metadata={"modality": "MR"},
+)
+
+augment = Compose([
+    RandomAffine(rotation=15, scale=(0.9, 1.1), p=0.7),
+    ElasticDeform(alpha=(120, 120, 10), sigma=(10, 10, 3), p=0.5),
+    BiasField(alpha=0.3, p=0.7),
+    OneOf([
+        RicianNoise(std=(0.005, 0.02)),   # MRI physical noise model
+        GaussianNoise(std=0.015),          # simpler additive noise
+    ], p=0.5),
+    GammaCorrection(gamma=(0.8, 1.2), p=0.5),
+], seed=42)
+
+out = augment(vol)           # mask kept in sync automatically
+print(out.image.shape, out.mask.shape)
+```
+
+Mask values are preserved exactly — spatial transforms use nearest-neighbour
+interpolation for the mask and the same random draw as the image.
+
+---
+
+## Preset pipelines
+
+Ready-to-use `Compose` pipelines for common modalities. All presets are
+seeded, deterministic, and serialisable.
+
+```python
+from medaugmentx.presets import mri_pipeline, ct_pipeline, dxr_pipeline, dbt_pipeline
+
+mri  = mri_pipeline(seed=42)   # bias field, Rician noise, ghosting/k-space
+ct   = ct_pipeline(seed=42)    # window/level, Gaussian noise, beam hardening
+dxr  = dxr_pipeline(seed=42)  # blur, brightness/contrast, low-resolution sim
+dbt  = dbt_pipeline(seed=42)  # slab shift, limited-angle blur, anisotropic elastic
+```
+
+Each preset is a starting point — tune the parameters or swap transforms for
+your dataset.
+
+---
+
+## Serialisation
+
+Pipelines serialise to JSON (built-in) or YAML (optional PyYAML):
+
+```python
+from medaugmentx.serialization import to_json, from_json
+
+pipeline = mri_pipeline(seed=42)
+
+# Save
+json_str = to_json(pipeline)
+with open("pipeline.json", "w") as f:
+    f.write(json_str)
+
+# Reload — exact same structure, ready to run
+pipeline2 = from_json(open("pipeline.json").read())
+out = pipeline2(vol)
+```
+
+Custom transforms can be added to the registry so they serialise too:
+
+```python
+from medaugmentx.serialization import REGISTRY
+REGISTRY["MyTransform"] = MyTransform
+```
+
+---
+
+## Loading real volumes
+
+```python
+from medaugmentx.io import load_dicom_series, load_nifti, save_nifti
+
+vol_ct  = load_dicom_series("/path/to/study/CT_chest/")  # MedVolume
+vol_mri = load_nifti("brain_t1.nii.gz")                   # MedVolume
+
+augmented = ct_pipeline(seed=0)(vol_ct)
+save_nifti(augmented, "ct_augmented.nii.gz")
+```
+
+Both loaders populate `spacing` and `metadata` (modality, vendor, DICOM tags)
+automatically.
+
+> Requires the `io` extra: `pip install "medaugmentx[io]"`
+
+---
+
+## What's available
+
+### Core
+
+| Component | Description |
+| --- | --- |
+| `MedVolume` | Container: image + optional mask + spacing + metadata |
+| `Transform` | Abstract base — probability gating, seedable RNG, `to_dict()` |
+| `Compose` | Sequential pipeline with deterministic child seeding |
+| `OneOf` / `SomeOf` | Random selection from a set of transforms |
+
+### Spatial transforms
+
+| Transform | What it does |
+| --- | --- |
+| `RandomAffine` | Rotation + scaling + translation, 2D/3D, axis-aware |
+| `RandomFlip` | Per-axis flip with independent probability |
+| `AnatomicCrop` | Foreground-biased random crop |
+| `ElasticDeform` | Anisotropic B-spline elastic deformation |
+
+### Intensity transforms
+
+| Transform | What it does |
+| --- | --- |
+| `GaussianNoise` | Additive zero-mean Gaussian noise |
+| `RicianNoise` | MRI magnitude noise (physically correct for low-SNR regions) |
+| `GammaCorrection` | Power-law contrast, normalisation-aware |
+| `BiasField` | Smooth multiplicative field (MRI coil inhomogeneity) |
+| `WindowLevel` | Random CT window/level shift (protocol variation) |
+| `BrightnessContrast` | Additive brightness + multiplicative contrast |
+| `GaussianBlur` | Isotropic Gaussian blur |
+| `SimulateLowResolution` | Downsample + upsample (cross-site resolution variation) |
+
+### Modality-specific — MRI
+
+| Transform | What it does |
+| --- | --- |
+| `GhostingArtifact` | Phase-encoding ghosting (shifted attenuated replica) |
+| `KSpaceDropout` | Random k-space line zeroing with Gibbs ringing reconstruction |
+
+### Modality-specific — CT
+
+| Transform | What it does |
+| --- | --- |
+| `BeamHardening` | Radially-symmetric cupping artifact |
+
+### Modality-specific — Tomosynthesis (DBT)
+
+| Transform | What it does |
+| --- | --- |
+| `SlabShift` | Z-axis reconstruction-centre variation |
+| `LimitedAngleBlur` | Arc-angle-dependent Z-only blur |
+| `SliceDropout` | Random slice zeroing (robustness) |
+| `AnisotropicElastic` | DBT-default elastic deformation |
+
+### I/O
+
+| Helper | Description |
+| --- | --- |
+| `load_dicom_series(path)` | 3D volume from a DICOM series directory |
+| `load_nifti(path)` | MedVolume from `.nii` / `.nii.gz` |
+| `save_nifti(vol, path)` | Write MedVolume to NIfTI |
+
+---
+
+## Reproducibility
+
+Every transform is seedable. `Compose(..., seed=42)` produces bit-identical
+output across runs and across machines (within a NumPy version):
+
+```python
+a = Compose([...], seed=42)(vol)
+b = Compose([...], seed=42)(vol)
+assert np.array_equal(a.image, b.image)  # always passes
+```
+
+---
+
+## Project status & roadmap
+
+| Phase | Status |
+| --- | --- |
+| **1 — Core MVP** | ✅ Core data model, spatial/intensity transforms, DBT, DICOM/NIfTI I/O |
+| **2 — Modality artifacts & serialisation** | ✅ MRI (bias field, ghosting, k-space), CT (beam hardening), presets, JSON/YAML |
+| **3 — GPU backend, framework interop, v1.0** | Planned |
+
+Detailed deliverables: [docs/MILESTONES.md](docs/MILESTONES.md).
+
+---
+
+## Contributing
+
+We welcome PRs — new transforms, vendor DICOM coverage, and bug reports with
+sample data are especially valuable. See [CONTRIBUTING.md](CONTRIBUTING.md)
+for the development setup, testing conventions, and the transform-authoring
+template.
+
+## License
+
+MIT — see [LICENSE](LICENSE).

medaugmentx-0.2.0/README.md

@@ -0,0 +1,273 @@
+# MedAugmentX
+
+**Clinically-aware medical image augmentation for AI training.**
+
+[![Python](https://img.shields.io/badge/python-3.9%20%E2%80%93%203.14-blue.svg)](https://www.python.org/downloads/)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](LICENSE)
+[![Status: Phase 2](https://img.shields.io/badge/status-phase%202-green.svg)](docs/MILESTONES.md)
+
+MedAugmentX is a purpose-built Python library for data augmentation of medical
+images. Unlike general-purpose libraries (`torchvision`, `albumentations`), it
+treats the unique properties of medical data — anisotropic 3D volumes,
+modality-specific artifacts, mask consistency, and clinical I/O — as
+first-class concerns.
+
+**Zero deep-learning dependencies.** The core library requires only NumPy and
+SciPy. PyTorch, MONAI, and TorchIO interop are planned for Phase 3.
+
+---
+
+## Why MedAugmentX
+
+| Problem with general-purpose augmentation | What MedAugmentX does |
+| --- | --- |
+| Treats every image as 2D RGB | Native 3D volumes with anisotropic-aware ops |
+| No MRI-specific noise or artifact models | Physics-based Rician noise, bias field, k-space corruption, ghosting |
+| CT/DXR augmentation is generic brightness shift | Window/level variation, beam hardening, resolution simulation |
+| Masks drift on complex transforms | Deterministic seeding; image and mask share the same random draw |
+| No tomosynthesis support | Dedicated DBT module with slab-aware augmentations |
+| DICOM/NIfTI I/O scattered across libraries | One loader, one `MedVolume`, vendor metadata preserved |
+| Pipelines can't be saved or reloaded | Full JSON/YAML serialisation with round-trip reconstruction |
+
+---
+
+## Install
+
+```bash
+# Core (NumPy + SciPy only)
+pip install medaugmentx
+
+# With DICOM/NIfTI I/O
+pip install "medaugmentx[io]"
+
+# With YAML serialisation support
+pip install "medaugmentx[yaml]"
+```
+
+Verify the installation:
+
+```python
+import medaugmentx
+print(medaugmentx.__version__)   # 0.2.0
+```
+
+---
+
+## Quick start
+
+### One-line preset
+
+The fastest way to get started is a pre-built modality preset:
+
+```python
+from medaugmentx.presets import mri_pipeline, ct_pipeline, dxr_pipeline, dbt_pipeline
+
+pipeline = mri_pipeline(seed=42)
+augmented = pipeline(vol)          # MedVolume in, MedVolume out
+```
+
+Four presets ship out of the box — see [Preset pipelines](#preset-pipelines) below.
+
+### Build your own pipeline
+
+```python
+import numpy as np
+from medaugmentx import MedVolume, Compose, OneOf
+from medaugmentx.transforms import (
+    RandomAffine, ElasticDeform,
+    BiasField, RicianNoise, GaussianNoise, GammaCorrection,
+)
+
+vol = MedVolume(
+    image=np.random.rand(80, 256, 256).astype(np.float32),
+    mask=np.zeros((80, 256, 256), dtype=np.uint8),
+    spacing=(1.0, 0.7, 0.7),          # mm per axis (z, y, x)
+    metadata={"modality": "MR"},
+)
+
+augment = Compose([
+    RandomAffine(rotation=15, scale=(0.9, 1.1), p=0.7),
+    ElasticDeform(alpha=(120, 120, 10), sigma=(10, 10, 3), p=0.5),
+    BiasField(alpha=0.3, p=0.7),
+    OneOf([
+        RicianNoise(std=(0.005, 0.02)),   # MRI physical noise model
+        GaussianNoise(std=0.015),          # simpler additive noise
+    ], p=0.5),
+    GammaCorrection(gamma=(0.8, 1.2), p=0.5),
+], seed=42)
+
+out = augment(vol)           # mask kept in sync automatically
+print(out.image.shape, out.mask.shape)
+```
+
+Mask values are preserved exactly — spatial transforms use nearest-neighbour
+interpolation for the mask and the same random draw as the image.
+
+---
+
+## Preset pipelines
+
+Ready-to-use `Compose` pipelines for common modalities. All presets are
+seeded, deterministic, and serialisable.
+
+```python
+from medaugmentx.presets import mri_pipeline, ct_pipeline, dxr_pipeline, dbt_pipeline
+
+mri  = mri_pipeline(seed=42)   # bias field, Rician noise, ghosting/k-space
+ct   = ct_pipeline(seed=42)    # window/level, Gaussian noise, beam hardening
+dxr  = dxr_pipeline(seed=42)  # blur, brightness/contrast, low-resolution sim
+dbt  = dbt_pipeline(seed=42)  # slab shift, limited-angle blur, anisotropic elastic
+```
+
+Each preset is a starting point — tune the parameters or swap transforms for
+your dataset.
+
+---
+
+## Serialisation
+
+Pipelines serialise to JSON (built-in) or YAML (optional PyYAML):
+
+```python
+from medaugmentx.serialization import to_json, from_json
+
+pipeline = mri_pipeline(seed=42)
+
+# Save
+json_str = to_json(pipeline)
+with open("pipeline.json", "w") as f:
+    f.write(json_str)
+
+# Reload — exact same structure, ready to run
+pipeline2 = from_json(open("pipeline.json").read())
+out = pipeline2(vol)
+```
+
+Custom transforms can be added to the registry so they serialise too:
+
+```python
+from medaugmentx.serialization import REGISTRY
+REGISTRY["MyTransform"] = MyTransform
+```
+
+---
+
+## Loading real volumes
+
+```python
+from medaugmentx.io import load_dicom_series, load_nifti, save_nifti
+
+vol_ct  = load_dicom_series("/path/to/study/CT_chest/")  # MedVolume
+vol_mri = load_nifti("brain_t1.nii.gz")                   # MedVolume
+
+augmented = ct_pipeline(seed=0)(vol_ct)
+save_nifti(augmented, "ct_augmented.nii.gz")
+```
+
+Both loaders populate `spacing` and `metadata` (modality, vendor, DICOM tags)
+automatically.
+
+> Requires the `io` extra: `pip install "medaugmentx[io]"`
+
+---
+
+## What's available
+
+### Core
+
+| Component | Description |
+| --- | --- |
+| `MedVolume` | Container: image + optional mask + spacing + metadata |
+| `Transform` | Abstract base — probability gating, seedable RNG, `to_dict()` |
+| `Compose` | Sequential pipeline with deterministic child seeding |
+| `OneOf` / `SomeOf` | Random selection from a set of transforms |
+
+### Spatial transforms
+
+| Transform | What it does |
+| --- | --- |
+| `RandomAffine` | Rotation + scaling + translation, 2D/3D, axis-aware |
+| `RandomFlip` | Per-axis flip with independent probability |
+| `AnatomicCrop` | Foreground-biased random crop |
+| `ElasticDeform` | Anisotropic B-spline elastic deformation |
+
+### Intensity transforms
+
+| Transform | What it does |
+| --- | --- |
+| `GaussianNoise` | Additive zero-mean Gaussian noise |
+| `RicianNoise` | MRI magnitude noise (physically correct for low-SNR regions) |
+| `GammaCorrection` | Power-law contrast, normalisation-aware |
+| `BiasField` | Smooth multiplicative field (MRI coil inhomogeneity) |
+| `WindowLevel` | Random CT window/level shift (protocol variation) |
+| `BrightnessContrast` | Additive brightness + multiplicative contrast |
+| `GaussianBlur` | Isotropic Gaussian blur |
+| `SimulateLowResolution` | Downsample + upsample (cross-site resolution variation) |
+
+### Modality-specific — MRI
+
+| Transform | What it does |
+| --- | --- |
+| `GhostingArtifact` | Phase-encoding ghosting (shifted attenuated replica) |
+| `KSpaceDropout` | Random k-space line zeroing with Gibbs ringing reconstruction |
+
+### Modality-specific — CT
+
+| Transform | What it does |
+| --- | --- |
+| `BeamHardening` | Radially-symmetric cupping artifact |
+
+### Modality-specific — Tomosynthesis (DBT)
+
+| Transform | What it does |
+| --- | --- |
+| `SlabShift` | Z-axis reconstruction-centre variation |
+| `LimitedAngleBlur` | Arc-angle-dependent Z-only blur |
+| `SliceDropout` | Random slice zeroing (robustness) |
+| `AnisotropicElastic` | DBT-default elastic deformation |
+
+### I/O
+
+| Helper | Description |
+| --- | --- |
+| `load_dicom_series(path)` | 3D volume from a DICOM series directory |
+| `load_nifti(path)` | MedVolume from `.nii` / `.nii.gz` |
+| `save_nifti(vol, path)` | Write MedVolume to NIfTI |
+
+---
+
+## Reproducibility
+
+Every transform is seedable. `Compose(..., seed=42)` produces bit-identical
+output across runs and across machines (within a NumPy version):
+
+```python
+a = Compose([...], seed=42)(vol)
+b = Compose([...], seed=42)(vol)
+assert np.array_equal(a.image, b.image)  # always passes
+```
+
+---
+
+## Project status & roadmap
+
+| Phase | Status |
+| --- | --- |
+| **1 — Core MVP** | ✅ Core data model, spatial/intensity transforms, DBT, DICOM/NIfTI I/O |
+| **2 — Modality artifacts & serialisation** | ✅ MRI (bias field, ghosting, k-space), CT (beam hardening), presets, JSON/YAML |
+| **3 — GPU backend, framework interop, v1.0** | Planned |
+
+Detailed deliverables: [docs/MILESTONES.md](docs/MILESTONES.md).
+
+---
+
+## Contributing
+
+We welcome PRs — new transforms, vendor DICOM coverage, and bug reports with
+sample data are especially valuable. See [CONTRIBUTING.md](CONTRIBUTING.md)
+for the development setup, testing conventions, and the transform-authoring
+template.
+
+## License
+
+MIT — see [LICENSE](LICENSE).

medaugmentx-0.2.0/medaugmentx/__init__.py

@@ -0,0 +1,22 @@
+"""MedAugment — clinically-aware medical image augmentation.
+
+Public surface for Phase 2.
+"""
+from medaugmentx.core import (
+    Compose,
+    MedVolume,
+    OneOf,
+    SomeOf,
+    Transform,
+)
+
+__version__ = "0.2.0"
+
+__all__ = [
+    "__version__",
+    "MedVolume",
+    "Transform",
+    "Compose",
+    "OneOf",
+    "SomeOf",
+]

medaugmentx-0.2.0/medaugmentx/core/__init__.py

@@ -0,0 +1,16 @@
+"""Core data model and pipeline primitives."""
+from medaugmentx.core.base import Transform
+from medaugmentx.core.compose import Compose, OneOf, SomeOf
+from medaugmentx.core.utils import as_float32, derive_rng, resolve_rng
+from medaugmentx.core.volume import MedVolume
+
+__all__ = [
+    "MedVolume",
+    "Transform",
+    "Compose",
+    "OneOf",
+    "SomeOf",
+    "as_float32",
+    "derive_rng",
+    "resolve_rng",
+]