Commit Graph

128 Commits

Author SHA1 Message Date
harisreedhar e8cc2bfff1 add gaze loss 2025-03-11 14:43:10 +01:00
henryruhs a951d700fc Restore RGB normalization 2025-03-11 14:43:10 +01:00
henryruhs 6eff69a41a There is no need to make directories 2025-03-11 14:43:10 +01:00
henryruhs 5b3b2abdd7 Remove annoying Tuner, Match both trainer configs a bit more 2025-03-11 14:43:10 +01:00
henryruhs bfcbd6bf95 Fix CI 2025-03-11 14:43:10 +01:00
henryruhs 607c55ff1f Remove Numpy and CV2 to fully use Tensors 2025-03-11 14:43:10 +01:00
henryruhs 257e5e56a4 Rename split ratio 2025-03-11 14:43:10 +01:00
henryruhs f19908ccd6 Fix spacing for lambda 2025-03-11 14:43:10 +01:00
henryruhs 18a2531b54 Fix naming in dataset.py 2025-03-11 14:43:10 +01:00
henryruhs 84be7d1ffb Remove deprecated argument 2025-03-11 14:43:10 +01:00
henryruhs 8b53c76a0a Remove deprecated argument 2025-03-11 14:43:10 +01:00
henryruhs 303cbfa024 Final refactoring for AAD done 2025-03-11 14:43:10 +01:00
henryruhs bc174186eb Final refactoring for AAD done 2025-03-11 14:43:10 +01:00
henryruhs d7158749c2 Improve naming 2025-03-11 14:43:10 +01:00
henryruhs 8b2b6892aa Improve 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 335d597e53 Rename to aienet.py 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 de6cfbc35b Fix CI 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 38211f0340 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
henryruhs 086d9eed87 Introduce new ReconstructionLoss class 2025-03-11 14:43:09 +01:00
henryruhs 085c493e18 Rename id_embedder to embedder, Tons of naming in training step, Introduce new IdentityLoss class 2025-03-11 14:43:09 +01:00
henryruhs f482d46798 Remove VisionTensor from Discriminator 2025-03-11 14:43:09 +01:00
henryruhs a6e1405c70 Restore map_location = 'cpu' 2025-03-11 14:43:09 +01:00
henryruhs ac41bab3a2 Restore map_location = 'cpu' 2025-03-11 14:43:09 +01:00
henryruhs 206a1411d1 Fix Generator 2025-03-11 14:43:09 +01:00
henryruhs 10ce04ed58 Introduce feature flag for Unet 2025-03-11 14:43:09 +01:00
henryruhs 094d5cea9e Fix CI 2025-03-11 14:43:09 +01:00
henryruhs f6c59257d9 Partial use Resnet34 as a DownSample replacement 2025-03-11 14:43:09 +01:00
henryruhs 83ef075b1d Remove map_location 2025-03-11 14:43:09 +01:00
henryruhs 00dccf07b9 Fix tensor foobar 2025-03-11 14:43:09 +01:00
henryruhs 5a6e3393e2 Revert loss for the moment 2025-03-11 14:43:09 +01:00
henryruhs c17378f3c7 Let's call it batch ratio 2025-03-11 14:43:09 +01:00
henryruhs 723e9fde78 follow the not invented here syndrome 2025-03-11 14:43:09 +01:00