Views
No views yet
nnUNetTrainerRegression_mae_deep, 3D full-resolution).| File | Description |
|---|---|
checkpoints/model.pt | TorchScript compiled model (~390 MB) |
checkpoints/metadata.json | Patch size, normalization stats, inference config |
sct_generator.py | Self-contained inference class — no nnUNet install required |
model.pt is a TorchScript module exported with torch.jit.save() and must be loaded with torch.jit.load().torch
scipy
numpy1import numpy as np
2from sct_generator import StandaloneRegressionInference
3
4# Load model (pass the directory containing model.pt and metadata.json)
5model = StandaloneRegressionInference(
6 model_path="checkpoints/",
7 device="cuda" # or "cpu"
8)
9
10# cbct_volume: 3D numpy array of HU values, shape (D, H, W)
11cbct_volume = np.load("your_cbct_volume.npy")
12
13# Run inference — returns sCT volume in HU, same spatial shape as input
14sct_output = model.predict(cbct_volume)predict() handles internally:| Field | Value |
|---|---|
| Trainer | nnUNetTrainerRegression_mae_deep |
| Configuration | 3d_fullres |
| Fold | all |
| Loss | MAE |
| Input | Physics-based simulated CBCT |
| Output | CT |
| Anatomy | Pelvis |
1@article{zimmermann2026simcbct,
2 title = {Eliminating Registration Bias in Synthetic CT Generation:
3 A Physics-Based Simulation Framework},
4 author = {Zimmermann, Lukas and Rauter, Michael and Schmid, Maximilian
5 and Georg, Dietmar and Kn\"{a}usl, Barbara},
6 journal = {arXiv preprint arXiv:2602.02130},
7 year = {2026}
8}