pyRadPlan-dosecalc-Det-VHEE-200-lung

pyRadPlan-dosecalc-Det-VHEE-200-lung

Deterministic VHEE (Very High Energy Electron) dose predictor for lung cases, trained for a single fixed beam energy of 200 MeV (ConvBayes_new architecture, single forward pass). Given a CT cuboid and a beamlet, predicts a local physical dose cuboid.

Usable directly as the AIBeamletEngine dose-calculation engine in pyRadPlan:

pln.prop_dose_calc = {"engine": "AIBeamlet", "model": "DKFZ-RadOpt/pyRadPlan-dosecalc-Det-VHEE-200-lung"}

VHEE plans use one fixed energy per beam. Companion models cover the other supported energies — pyRadPlan-dosecalc-Det-VHEE-100-lung and pyRadPlan-dosecalc-Det-VHEE-150-lung; load the one that matches pln.prop_stf["energy"].

Repository contents

Follows the pyRadPlan model contract (pyRadPlan.ai.modelhub):

Outputs

• physical_dose — predicted physical dose

Training assumptions

• Radiation mode: VHEE (Very High Energy Electrons)
• Beam energy: 200 MeV (fixed — single-energy model)
• Anatomy: lung
• Trained machine: Generic
• Sampling grid: 2 mm spacing, 72 mm lateral field width (±36 mm), depth window [−350, 350] mm relative to the beamlet origin

Predictions outside this range are not guaranteed to be accurate; AIBeamletEngine warns when the plan falls outside the declared range.

Security note: loading this model executes model.py/preprocessor.py shipped in this repository. pyRadPlan.ai.modelhub.load_model requires an explicit opt-in for hub models (trust_remote_code=True or PYRADPLAN_AI_MODELHUB_TRUST_REMOTE_CODE=1); AIBeamletEngine runs the downloaded copy directly.