Commit Graph

Select branches

Hide Pull Requests

master

#1

#10

#11

#12

#13

#14

#15

#16

#17

#18

#19

#2

#20

#21

#22

#23

#24

#25

#26

#27

#28

#29

#3

#30

#31

#32

#33

#34

#35

#36

#37

#38

#39

#4

#40

#41

#42

#43

#44

#45

#46

#47

#48

#49

#5

#50

#51

#52

#53

#54

#55

#56

#57

#58

#59

#6

#60

#61

#62

#63

#64

#65

#66

#67

#68

#69

#7

#70

#71

#72

#73

#74

#75

#76

#77

#78

#79

#8

#80

#81

#82

#83

#84

#85

#86

#87

#88

#89

#9

#90

#91

#92

#93

#94

#95

#96

models

Mono Color

• de6cfbc35b Fix CI henryruhs 2025-02-23 01:06:18 +01:00
• ed0f6ae897 Use new loss code, Remove unused code, Remove old types, Ban VisionTensor naming henryruhs 2025-02-23 01:05:01 +01:00
• 63e4bea3cd Use new config for old code henryruhs 2025-02-23 00:39:48 +01:00
• 14b9bccafe Introduce new DiscriminatorLoss class, Remove useless super call params henryruhs 2025-02-23 00:29:41 +01:00
• 579d3ef51c Introduce new GazeLoss class (switched to mean) henryruhs 2025-02-22 23:48:50 +01:00
• a797548329 Introduce new PoseLoss class (switched to mean) henryruhs 2025-02-22 23:15:39 +01:00
• 6eabcad1d0 Introduce new AttributeLoss class henryruhs 2025-02-22 22:31:44 +01:00
• 7848d28b02 Introduce new AdversarialLoss class fix henryruhs 2025-02-22 16:59:28 +01:00
• 30e787129a Introduce new AdversarialLoss class henryruhs 2025-02-22 16:28:00 +01:00
• 38211f0340 Introduce new AdversarialLoss class henryruhs 2025-02-22 16:26:40 +01:00
• f2833a32c3 Introduce new AdversarialLoss class henryruhs 2025-02-22 16:24:31 +01:00
• 3b7d3b6688 Introduce new ReconstructionLoss class henryruhs 2025-02-22 14:30:52 +01:00
• 086d9eed87 Introduce new ReconstructionLoss class henryruhs 2025-02-22 14:29:16 +01:00
• 085c493e18 Rename id_embedder to embedder, Tons of naming in training step, Introduce new IdentityLoss class henryruhs 2025-02-22 12:37:47 +01:00
• 2220f5ef08 Add missing encoder_type to config henryruhs 2025-02-23 00:31:31 +01:00
• f482d46798 Remove VisionTensor from Discriminator henryruhs 2025-02-22 12:19:44 +01:00
• a6e1405c70 Restore map_location = 'cpu' henryruhs 2025-02-22 11:26:35 +01:00
• ac41bab3a2 Restore map_location = 'cpu' henryruhs 2025-02-22 11:21:34 +01:00
• 206a1411d1 Fix Generator henryruhs 2025-02-22 10:43:08 +01:00
• dc0abff0ce Minor typo henryruhs 2025-02-22 10:32:54 +01:00
• 10ce04ed58 Introduce feature flag for Unet henryruhs 2025-02-22 10:08:23 +01:00
• 094d5cea9e Fix CI henryruhs 2025-02-16 23:18:49 +01:00
• f6c59257d9 Partial use Resnet34 as a DownSample replacement henryruhs 2025-02-16 23:12:23 +01:00
• 83ef075b1d Remove map_location henryruhs 2025-02-22 09:40:47 +01:00
• 575f215408 lr only in optimizer henryruhs 2025-02-21 18:40:11 +01:00
• d153c68813 Better split_ratio henryruhs 2025-02-21 17:16:11 +01:00
• 00dccf07b9 Fix tensor foobar henryruhs 2025-02-21 17:01:22 +01:00
• 5a6e3393e2 Revert loss for the moment henryruhs 2025-02-21 16:53:32 +01:00
• c17378f3c7 Let's call it batch ratio henryruhs 2025-02-21 16:38:53 +01:00
• 84503761b9 follow the not invented here syndrome henryruhs 2025-02-21 16:30:27 +01:00
• 723e9fde78 follow the not invented here syndrome henryruhs 2025-02-21 16:29:26 +01:00
• 09e913233b Follow convention of the other project henryruhs 2025-02-21 16:23:54 +01:00
• b4bbd862e2 Follow convention of the other project henryruhs 2025-02-21 16:13:13 +01:00
• 4d8433f54a changes harisreedhar 2025-02-21 20:12:48 +05:30
• db44c91dd8 changes harisreedhar 2025-02-21 19:40:44 +05:30
• b33281425a This is not a TensorDataset henryruhs 2025-02-21 11:03:31 +01:00
• 3d9ff4add0 Adjust validation to one time per epoch henryruhs 2025-02-21 10:26:07 +01:00
• 5934b47961 Add cosine_similarity henryruhs 2025-02-21 10:15:56 +01:00
• 04eaa831ea Rename to resume_path only henryruhs 2025-02-21 10:06:07 +01:00
• c1bed34c27 More tweaks henryruhs 2025-02-21 09:45:44 +01:00
• 4078681031 Fix typing henryruhs 2025-02-21 09:17:59 +01:00
• 15ee6fa763 Simplify sizes henryruhs 2025-02-21 09:11:43 +01:00
• 251e610f0e Follow the lightning naming and call this dataset, Improve config and types henryruhs 2025-02-21 09:07:58 +01:00
• da51c5336d This can be marked static henryruhs 2025-02-21 08:39:08 +01:00
• 9bd68c3d14 Similarity validation for embedding converter henryruhs 2025-02-21 08:29:03 +01:00
• 0a50e2d706 Rename data loader henryruhs 2025-02-21 08:26:22 +01:00
• 40dcef7fc7 Modernize data loader, remove read image helper henryruhs 2025-02-21 01:16:46 +01:00
• c041073953 Modernize data loader, remove read image helper henryruhs 2025-02-21 01:15:28 +01:00
• 354315502b Modernize data loader, remove read image helper henryruhs 2025-02-21 01:14:44 +01:00
• 254bc17c98 Modernize data loader, remove read image helper henryruhs 2025-02-21 01:13:37 +01:00
• 39c0313202 Modernize data loader, remove read image helper henryruhs 2025-02-21 01:11:28 +01:00
• 3cf9711df0 Modernize data loader, remove read image helper henryruhs 2025-02-21 01:09:57 +01:00
• d1bf54276d changes harisreedhar 2025-02-20 21:14:54 +05:30
• b47c6b72ee changes harisreedhar 2025-02-20 18:09:37 +05:30
• dcf19634d1 Brain fart henryruhs 2025-02-19 09:39:06 +01:00
• 857365770f Make validation step more solid, failed on empty checksums henryruhs 2025-02-19 09:14:57 +01:00
• d25f2865a9 Normalize validation output henryruhs 2025-02-19 08:40:50 +01:00
• 0d45568bd1 Uniform validation, Add cosine_similarity validation to face swapper henryruhs 2025-02-18 14:17:52 +01:00
• ccf6fa7f43 Uniform validation, Add cosine_similarity validation to face swapper henryruhs 2025-02-18 14:16:47 +01:00
• 28977d37d6 Uniform validation, Add cosine_similarity validation to face swapper henryruhs 2025-02-18 14:14:23 +01:00
• c161da2f25 Adjust naming and typing henryruhs 2025-02-19 08:14:18 +01:00
• 39818a16df Fix CI henryruhs 2025-02-18 11:05:58 +01:00
• e1ba81f220 Improve generator namings, Flip args to source then target henryruhs 2025-02-18 11:01:33 +01:00
• bf696be097 Improve generator namings, Flip args to source then target henryruhs 2025-02-18 11:00:30 +01:00
• f63bc788ac Fix typo henryruhs 2025-02-18 09:43:24 +01:00
• 11bb9065ba Uniform resume checkpoint approach henryruhs 2025-02-17 23:58:33 +01:00
• 7b2b8f0f85 fix discriminator training harisreedhar 2025-02-18 10:59:43 +05:30
• 999f2c9cbe Rename variable of EmbeddingConverterTrainer henryruhs 2025-02-17 23:25:19 +01:00
• 254f3efe68 Remove live portrait submodule henryruhs 2025-02-17 00:08:40 +01:00
• 5bb41ecbb2 Adjust README henryruhs 2025-02-17 09:15:00 +01:00
• 777a8384c2 Extend README with TensorBoard guide henryruhs 2025-02-16 12:45:37 +01:00
• 0949c1358b Fix CI henryruhs 2025-02-16 11:58:51 +01:00
• d9e10a9f7c Use lightning over pytorch_lightning import, Configure tensorboard henryruhs 2025-02-16 11:56:01 +01:00
• cd4b10c832 Fix imports and update dependencies henryruhs 2025-02-16 10:54:01 +01:00
• 0b7d25a36e Adjust licenses henryruhs 2025-02-15 12:53:00 +01:00
• 026bcf0c97 Adjust licenses henryruhs 2025-02-15 12:51:40 +01:00
• aa5094e576 We need MS license henryruhs 2025-02-15 12:23:43 +01:00
• 8f8dfecdbc some fixes harisreedhar 2025-02-15 19:05:38 +05:30
• 030d912c1b some fixes harisreedhar 2025-02-15 19:04:43 +05:30
• 0e148845af Extend gitignore henryruhs 2025-02-14 19:52:19 +01:00
• e1e0c11bb5 Fix CI henryruhs 2025-02-14 16:40:39 +01:00
• 88c4e53192 Use skip_tensor variable henryruhs 2025-02-14 16:38:25 +01:00
• 650551c19b Simplify Batch type henryruhs 2025-02-14 16:19:14 +01:00
• a971506271 Revert changes in generator henryruhs 2025-02-14 16:12:15 +01:00
• b69f69d015 Fix UNet henryruhs 2025-02-14 16:10:39 +01:00
• 953525e6b0 Fix typo henryruhs 2025-02-14 10:21:33 +01:00
• 58ad6af619 Add Sample suffix again henryruhs 2025-02-14 09:52:51 +01:00
• 7264884ff9 Fix CI henryruhs 2025-02-13 19:13:49 +01:00
• 1872f99584 Refacto UNet henryruhs 2025-02-13 19:11:20 +01:00
• 29e82f909a Move nld to networks henryruhs 2025-02-13 21:29:56 +01:00
• 257ab668ee Name what it is henryruhs 2025-02-13 19:02:00 +01:00
• 62d897f9d8 Direct return layers henryruhs 2025-02-13 18:59:30 +01:00
• f05ff6cdb1 Enforce IR version for older onnxruntime henryruhs 2025-02-13 12:52:59 +01:00
• 3fe32d7832 Fix example config henryruhs 2025-02-13 12:37:57 +01:00
• 3be8368eaa Improve lot of types, imports and names henryruhs 2025-02-12 21:26:57 +01:00
• b6b4f9f65b Improve lot of types, imports and names henryruhs 2025-02-12 21:25:43 +01:00
• e33bc0d52a Modernize to use ModuleList, Fix some types henryruhs 2025-02-12 17:15:15 +01:00
• f3409c5ade Modernize to use ModuleList, Fix some types henryruhs 2025-02-12 17:08:12 +01:00
• c6d16c0cf6 Modernize to use ModuleList, Fix some types henryruhs 2025-02-12 17:06:20 +01:00
• dfe7ab3b6f Modernize to use ModuleList, Fix some types henryruhs 2025-02-12 17:05:37 +01:00