Commit Graph

83 Commits

Author SHA1 Message Date
henryruhs afab997ffc Rename parser to face parser 2025-03-11 14:43:10 +01:00
henryruhs e758eb3e19 Use sequential whenever possible 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 dbe931e950 Fix naming 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 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
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 e9ea9cd9e5 Clean GazeLoss 2025-03-11 14:43:10 +01:00
harisreedhar 7cf5609c1f changes 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 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
harisreedhar df5895e266 changes 2025-03-11 14:43:10 +01:00
harisreedhar f2d3f8a19f changes 2025-03-11 14:43:10 +01:00
harisreedhar dfd9e99aed changes 2025-03-11 14:43:10 +01:00
henryruhs a22adaf51f Adjust some namings 2025-03-11 14:43:10 +01:00
henryruhs 0055c0c97f Adjust some namings 2025-03-11 14:43:10 +01:00
harisreedhar ea1b0205f0 constructor injection 2025-03-11 14:43:10 +01:00
harisreedhar d87f6c0b15 changes 2025-03-11 14:43:10 +01:00
harisreedhar 5d1b90ff19 changes 2025-03-11 14:43:10 +01:00
harisreedhar 2ddcf52b66 changes 2025-03-11 14:43:10 +01:00
henryruhs cadbe9cf76 Follow the concept of layers and sequences 2025-03-11 14:43:10 +01:00
henryruhs 7ce9d27097 Different naming 2025-03-11 14:43:10 +01:00
henryruhs 5bba2a1c69 Remove the condition from reconstruction loss 2025-03-11 14:43:10 +01:00
henryruhs 94480e16eb Rename to AAD as this is not a full AIENet 2025-03-11 14:43:10 +01:00
henryruhs bbcb1c35f0 Modernize AIENet 2025-03-11 14:43:10 +01:00
henryruhs ee3fc40e83 Move output channels to config 2025-03-11 14:43:10 +01:00
henryruhs 14bbece850 Rename id_ to identity_ 2025-03-11 14:43:10 +01:00
henryruhs fad38da864 Rename temp1 with positive and temp2 with negative 2025-03-11 14:43:10 +01:00
henryruhs e75a3c58f9 Replace calc() with forward(), Rename temp1 with positive and temp2 with negative 2025-03-11 14:43:10 +01:00
harisreedhar 6fed877d33 reconstruction-loss fix 2025-03-11 14:43:10 +01:00
henryruhs ed0f6ae897 Use new loss code, Remove unused code, Remove old types, Ban VisionTensor naming 2025-03-11 14:43:10 +01:00
henryruhs 63e4bea3cd Use new config for old code 2025-03-11 14:43:10 +01:00
henryruhs 14b9bccafe Introduce new DiscriminatorLoss class, Remove useless super call params 2025-03-11 14:43:10 +01:00
henryruhs 579d3ef51c Introduce new GazeLoss class (switched to mean) 2025-03-11 14:43:10 +01:00
henryruhs a797548329 Introduce new PoseLoss class (switched to mean) 2025-03-11 14:43:10 +01:00
henryruhs 6eabcad1d0 Introduce new AttributeLoss class 2025-03-11 14:43:10 +01:00
henryruhs 7848d28b02 Introduce new AdversarialLoss class fix 2025-03-11 14:43:10 +01:00
henryruhs 30e787129a Introduce new AdversarialLoss class 2025-03-11 14:43:09 +01:00
henryruhs f2833a32c3 Introduce new AdversarialLoss class 2025-03-11 14:43:09 +01:00
henryruhs 3b7d3b6688 Introduce new ReconstructionLoss class 2025-03-11 14:43:09 +01:00