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
..
byol	Various changes to fix testing	2021-06-11 15:31:10 -06:00
classifiers	Various mods to support better jpeg image filtering	2021-06-25 13:16:15 -06:00
diffusion	Add unet with latent guide	2021-06-26 11:02:58 -06:00
fixup_resnet	More refactoring	2020-12-18 09:18:34 -07:00
flownet2@db2b7899ea	Update flownet submodule	2020-10-24 11:59:00 -06:00
glean	Glean mods	2020-12-27 12:25:06 -07:00
global_convs	Misc	2021-04-21 18:09:46 -06:00
optical_flow	Add PWCNet for human optical flow	2021-01-25 08:25:44 -07:00
pixel_level_contrastive_learning	resnet_unet_3	2021-01-15 14:51:03 -07:00
segformer	Also debug distribution of switch	2021-06-07 15:36:07 -06:00
srflow	Migrate generators to dynamic model registration	2020-12-24 23:02:10 -07:00
styled_sr	Styled SR fixes	2021-01-05 20:14:39 -07:00
stylegan	misc adjustments for stylegan	2021-04-21 18:14:17 -06:00
switched_conv	Another fix to anorm	2021-06-06 15:09:49 -06:00
tecogan	Migrate generators to dynamic model registration	2020-12-24 23:02:10 -07:00
transformers	Injector auto-registration	2020-12-29 20:58:02 -07:00
vqvae	Fix some bugs, checkin work on vqvae3	2021-03-02 20:56:19 -07:00
__init__.py	Lots of new discriminator nets	2020-11-10 16:06:54 -07:00
arch_util.py	Change default ReLU slope to .2 BREAKS COMPATIBILITY	2020-12-19 08:28:03 -07:00
discriminator_vgg_arch.py	Misc fixes	2021-01-04 10:53:53 -07:00
feature_arch.py	More refactoring	2020-12-18 09:18:34 -07:00
lightweight_gan.py	Mods to support lightweight_gan model	2021-03-02 20:51:48 -07:00
ResGen_arch.py	More refactoring	2020-12-18 09:18:34 -07:00
RRDBNet_arch.py	Move switched_conv logic around a bit	2021-02-02 20:41:24 -07:00
spinenet_arch.py	Migrate generators to dynamic model registration	2020-12-24 23:02:10 -07:00