Commit Graph

273 Commits

Author SHA1 Message Date
henryruhs b59e172fa3 Always config injection 2025-03-11 14:43:10 +01:00
henryruhs 368da824aa Fix CI 2025-03-11 14:43:10 +01:00
henryruhs e61e470432 Fix CI 2025-03-11 14:43:10 +01:00
henryruhs c8953ce8a1 Apply new config approach for embedding converter 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 aedaa20d78 Remove UnetPro, make float values visible in README 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 f791178ded Fix CI 2025-03-11 14:43:10 +01:00
henryruhs 866019d44f Debug path 2025-03-11 14:43:10 +01:00
henryruhs 94ad33cb1e Debug path 2025-03-11 14:43:10 +01:00
henryruhs de72e50233 Only ubuntu needed for test 2025-03-11 14:43:10 +01:00
henryruhs 5f053f9f69 Crazy fix 2025-03-11 14:43:10 +01:00
henryruhs f678aa8f7e Crazy fix 2025-03-11 14:43:10 +01:00
henryruhs 0e8207ccc8 Crazy fix 2025-03-11 14:43:10 +01:00
henryruhs 4a2559c866 Crazy fix 2025-03-11 14:43:10 +01:00
henryruhs 3c5554c1c5 Crazy fix 2025-03-11 14:43:10 +01:00
henryruhs dbf5687bcd Crazy fix 2025-03-11 14:43:10 +01:00
henryruhs 2148e9b701 Prepare test for 512 2025-03-11 14:43:10 +01:00
henryruhs 64ebfa7b84 Add basic test for aad and unet 2025-03-11 14:43:10 +01:00
henryruhs daeec46e36 Add basic test for aad and unet 2025-03-11 14:43:10 +01:00
henryruhs 13d15029b7 Add basic test for aad and unet 2025-03-11 14:43:10 +01:00
henryruhs d5c51a90e8 Add basic test for aad and unet 2025-03-11 14:43:10 +01:00
henryruhs 6fa8d6b6eb Add basic test for aad and unet 2025-03-11 14:43:10 +01:00
henryruhs 72371b9f11 Add basic test for aad and unet 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 18a605e1a3 Change the weight config order 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 dd0a2fe649 Add albumentations 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 d93daf9e5f Introduce batch mode via config for equal and same batches 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