Barnabiii's picture
Update README.md
c0fa984 verified

HDUNet PBS

Trained by Margerie Huet Dastarac .
Training date: 06/06/2023 .

1. Task Description

Prediction of the dose distribution with proton therapy beam scanning treatment.

2. Model

2.1. Architecture

error while loading image

Figure 1: HDUNet PBS architecture

2.2. Input

  • CT
  • Target volumes contours and prescription
  • Organs at risks contours

2.3. Output

  • DOSE

2.4 Training details

  • Number of epoch: 400
  • Loss function: MSE loss
  • Optimizer: AdamW
  • Learning Rate: 0.0001
  • Dropout: No
  • Patch size in voxels: (128,128,128)
  • Data augmentation used:
    • RandCrop

3. Dataset

  • Location: Oropharynx
  • Training set size: 57
  • Resolution in mm: 3x3x3

Performance

  • TBD