Commit Graph

211 Commits

Author SHA1 Message Date
harisreedhar 606bf42089 fix 2025-03-11 14:43:10 +01:00
harisreedhar a3ac4d5ddd changes 2025-03-11 14:43:10 +01:00
harisreedhar 1659805b08 changes 2025-03-11 14:43:10 +01:00
harisreedhar 8f1f002c64 add masknet 2025-03-11 14:43:10 +01:00
harisreedhar 4af22832db fix discriminator & restore grad for ID embedder 2025-03-11 14:43:10 +01:00
henryruhs 9ff30a0268 Use getint 2025-03-11 14:43:10 +01:00
henryruhs dbe931e950 Fix naming 2025-03-11 14:43:10 +01:00
henryruhs 862dce7bc6 Revert the config dicts 2025-03-11 14:43:10 +01:00
henryruhs 8101b15e1c Revert the config dicts 2025-03-11 14:43:10 +01:00
henryruhs 1dfd230fc5 Revert the config dicts 2025-03-11 14:43:10 +01:00
henryruhs 7e938c2ec9 Use a lot of ignores 2025-03-11 14:43:10 +01:00
henryruhs c7d55d0d17 Use a lot of ignores 2025-03-11 14:43:10 +01:00
henryruhs 6bc44ad3d8 Fix CI 2025-03-11 14:43:10 +01:00
henryruhs b829d5e42c Migrate most to self.config and self.context 2025-03-11 14:43:10 +01:00
henryruhs ab3b699124 Rework on config 2025-03-11 14:43:10 +01:00
henryruhs bfa9924b40 Remove types 2025-03-11 14:43:10 +01:00
henryruhs a2a9b78dac More adjustments 2025-03-11 14:43:10 +01:00
henryruhs d944d95bfd Fix CI 2025-03-11 14:43:10 +01:00
henryruhs 8f1f63f2ef Fix CI 2025-03-11 14:43:10 +01:00
henryruhs 368da824aa Fix CI 2025-03-11 14:43:10 +01:00
harisreedhar e428ae04e3 add 128px 2025-03-11 14:43:10 +01:00
harisreedhar a89e51c2f8 remove grad for inference only models 2025-03-11 14:43:10 +01:00
harisreedhar 61f48d9246 changes 2025-03-11 14:43:10 +01:00
henryruhs dfd018a897 Fix CI 2025-03-11 14:43:10 +01:00
henryruhs b63562abad Simplify infer() 2025-03-11 14:43:10 +01:00
henryruhs e9ea9cd9e5 Clean GazeLoss 2025-03-11 14:43:10 +01:00
henryruhs abdc770892 Improve UNet 2025-03-11 14:43:10 +01:00
harisreedhar 7cf5609c1f changes 2025-03-11 14:43:10 +01:00
henryruhs 6388727262 Fix CI 2025-03-11 14:43:10 +01:00
henryruhs f9de4ce78a Remove UnetPro, make float values visible in README 2025-03-11 14:43:10 +01:00
henryruhs 6d805438ad Adjust networks for 512 2025-03-11 14:43:10 +01:00
henryruhs 7cc893c32e Adjust networks for 512 2025-03-11 14:43:10 +01:00
henryruhs 3e69d5a9a9 Adjust networks for 512 2025-03-11 14:43:10 +01:00
henryruhs 786adf73a2 Fix UnetPro 2025-03-11 14:43:10 +01:00
henryruhs dcc5ccccd7 Extend Unet with more layers 2025-03-11 14:43:10 +01:00
henryruhs 5056b8df75 Variable AAD layer according to output size 2025-03-11 14:43:10 +01:00
henryruhs 430c71d031 Adjust config and namings 2025-03-11 14:43:10 +01:00
harisreedhar 176dced1f6 changes 2025-03-11 14:43:10 +01:00
henryruhs fea75ff949 Remove useless comma 2025-03-11 14:43:10 +01:00
harisreedhar df5895e266 changes 2025-03-11 14:43:10 +01:00
harisreedhar f2d3f8a19f changes 2025-03-11 14:43:10 +01:00
henryruhs ceb3c0cfdf Rename to AugmentTransform 2025-03-11 14:43:10 +01:00
henryruhs e5a4a54e61 Introduce Transform classes, Add albumentations 2025-03-11 14:43:10 +01:00
henryruhs 9dc1031fa5 Fix same and equal 2025-03-11 14:43:10 +01:00
henryruhs 3a61da8bab Introduce batch mode via config for equal and same batches 2025-03-11 14:43:10 +01:00
harisreedhar dfd9e99aed changes 2025-03-11 14:43:10 +01:00
harisreedhar 2fb0b4289d changes 2025-03-11 14:43:10 +01:00
harisreedhar e6ea454360 supress warning 2025-03-11 14:43:10 +01:00
henryruhs 80d1694e23 Replace naming warp_matrix with warp_template 2025-03-11 14:43:10 +01:00
henryruhs 5e68de9170 Add config to dataloader alignment 2025-03-11 14:43:10 +01:00