DL-Art-School/codes/models/diffusion
James Betker 46e9f62be0 Add unet with latent guide
This is a diffusion network that uses both a LQ image
and a reference sample HQ image that is compressed into
a latent vector to perform upsampling

The hope is that we can steer the upsampling network
with sample images.
2021-06-26 11:02:58 -06:00
..
__init__.py Support gaussian diffusion models 2021-06-02 21:47:32 -06:00
fp16_util.py Support diffusion unet 2021-06-06 13:57:22 -06:00
gaussian_diffusion.py A few fixes: 2021-06-05 13:40:32 -06:00
losses.py Support gaussian diffusion models 2021-06-02 21:47:32 -06:00
nn.py Support gaussian diffusion models 2021-06-02 21:47:32 -06:00
resample.py Support gaussian diffusion models 2021-06-02 21:47:32 -06:00
respace.py GD mods & fixes 2021-06-04 17:13:16 -06:00
rrdb_diffusion.py Misc fixes for diffusion nets 2021-06-21 10:38:07 -06:00
unet_diffusion.py Pass a corruption factor through the dataset into the upsampling network 2021-06-07 09:13:54 -06:00
unet_latent_guide.py Add unet with latent guide 2021-06-26 11:02:58 -06:00