joh-schb/PlantLDM

Implement / Add Denoising U-Net

Closed this issue · 1 comments

Denoising U-Net that maps $\mathbb{R}^{H \times W}$ to $\mathbb{R}^{H \times W}$. Maybe use a pre-implemented one (e.g. code).

  • Implement normal UNet with Down- and Up-sampling
  • Positional Embedding for time conditioning
  • Time conditioning (maybe with FiLM layer)
  • Class conditioning

Class conditioning maybe on a later stage, if unconditional generation is successful.