search

lucidrains / med-seg-diff-pytorch

Implementation of MedSegDiff in Pytorch - SOTA medical segmentation using DDPM and filtering of features in fourier space
MIT License
199 stars 29 forks source link

Adding sampler with option to use ensembles like in the paper #9

Closed bf2harven closed 1 year ago

bf2harven commented 1 year ago
  1. added sample.py to sample from a trained model.
  2. --num_ens sets the number of ensembles. --save_uncertainty to save the std of the ensembles.
lucidrains commented 1 year ago

🙏