Commit Graph
57 Commits
Author SHA1 Message Date
henryruhs e758eb3e19 Use sequential whenever possible 2025-03-11 14:43:10 +01:00
henryruhs 7f16d0a10e Follow sequences pattern 2025-03-11 14:43:10 +01:00
henryruhs 54923abf7f Follow layers and sequences pattern 2025-03-11 14:43:10 +01:00
harisreedharandhenryruhs e267f9ffd5 reduce layer 2025-03-11 14:43:10 +01:00
harisreedharandhenryruhs c2a639229f changes 2025-03-11 14:43:10 +01:00
henryruhs f9d105ea2b Some code review 2025-03-11 14:43:10 +01:00
harisreedharandhenryruhs 606bf42089 fix 2025-03-11 14:43:10 +01:00
harisreedharandhenryruhs 1659805b08 changes 2025-03-11 14:43:10 +01:00
harisreedharandhenryruhs 8f1f002c64 add masknet 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 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
harisreedharandhenryruhs e428ae04e3 add 128px 2025-03-11 14:43:10 +01:00
henryruhs abdc770892 Improve UNet 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
harisreedharandhenryruhs 176dced1f6 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 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 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 ed0f6ae897 Use new loss code, Remove unused code, Remove old types, Ban VisionTensor naming 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 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
harisreedharandhenryruhs 030d912c1b some fixes 2025-03-11 14:43:09 +01:00
henryruhs 88c4e53192 Use skip_tensor variable 2025-03-11 14:43:09 +01:00
henryruhs b69f69d015 Fix UNet 2025-03-11 14:43:09 +01:00
henryruhs 953525e6b0 Fix typo 2025-03-11 14:43:09 +01:00
henryruhs 58ad6af619 Add Sample suffix again 2025-03-11 14:43:09 +01:00
henryruhs 7264884ff9 Fix CI 2025-03-11 14:43:09 +01:00