Commit Graph

52 Commits

Author SHA1 Message Date
henryruhs 32dfdcf1b3 Remove unsued 2025-03-11 14:43:09 +01:00
henryruhs 494b84aecb Refactor discriminator to use ModuleList, Reduce complexity of layer creation 2025-03-11 14:43:09 +01:00
henryruhs 860771e482 Fix weight identity 2025-03-11 14:43:09 +01:00
henryruhs b42d2b06e7 Improve generate_preview 2025-03-11 14:43:09 +01:00
henryruhs 11c038cb81 Fix monitor 2025-03-11 14:43:09 +01:00
henryruhs 746cc86d52 Fix monitor 2025-03-11 14:43:09 +01:00
henryruhs 58c81cd646 Move magic methods up 2025-03-11 14:43:09 +01:00
henryruhs 4d2038d4ce Rename loss_id to loss_identity 2025-03-11 14:43:09 +01:00
henryruhs 67ad9badac Small typo 2025-03-11 14:43:09 +01:00
henryruhs 1f4405be44 Fix CI 2025-03-11 14:43:09 +01:00
henryruhs 1b6e7a6ca5 Rename ArcFace Converter to Embedding Converter, Add EmbeddingDataset, Add learning rate to config 2025-03-11 14:43:06 +01:00
henryruhs 62a69cddd2 Fix import order 2025-03-11 14:43:03 +01:00
henryruhs 611618e413 Make Embedding great again 2025-03-11 14:43:03 +01:00
harisreedhar 6381e755d7 list join 2025-03-11 14:43:03 +01:00
harisreedhar 2ed558a873 cleanup 2025-03-11 14:43:03 +01:00
henryruhs b7e2d3ccd7 It should be models and networks 2025-03-11 14:43:03 +01:00
harisreedhar 66af7d7957 add face swapper preview 2025-03-11 14:43:03 +01:00
harisreedhar dc0ef53668 add face swapper preview 2025-03-11 14:43:03 +01:00
henryruhs a17d050648 Mixed bag of cleanups 2025-03-11 14:43:03 +01:00
henryruhs 989a81c751 Mixed bag of cleanups 2025-03-11 14:43:03 +01:00
henryruhs a1cd025f81 Mixed bag of cleanups 2025-03-11 14:43:03 +01:00
henryruhs 9c15f584aa Mixed bag of cleanups 2025-03-11 14:43:03 +01:00
henryruhs 5892460c3d Add the final license 2025-03-11 14:43:03 +01:00
harisreedhar bb0e3b4a8a remove liveportrait and update requirements 2025-03-11 14:43:03 +01:00
harisreedhar fcb3390796 again again cleaning 2025-03-11 14:43:03 +01:00
harisreedhar 3b8b6442fc add infer and some cleaning 2025-03-11 14:43:03 +01:00
harisreedhar cfcf0ee2bd add glob image pattern to config 2025-03-11 14:43:03 +01:00
harisreedhar 4260bd6c28 fix type 2025-03-11 14:43:03 +01:00
harisreedhar d018dd0633 some cleaning and add exporting 2025-03-11 14:43:03 +01:00
harisreedhar 798ba48a52 clean data_loader 2025-03-11 14:43:03 +01:00
harisreedhar e45f46d355 cleaning 2025-03-11 14:43:03 +01:00
harisreedhar 008a221f55 cleaning 2025-03-11 14:43:03 +01:00
henryruhs 10826558b4 Fix indent to tabs 2025-03-11 14:43:03 +01:00
henryruhs 23ac63d55b Fix more typing 2025-03-11 14:43:03 +01:00
henryruhs 2bbba3563b Use mypy and flake8 for face swapper 2025-03-11 14:43:03 +01:00
henryruhs e3273221c9 Delete unused file 2025-03-11 14:43:03 +01:00
harisreedhar 9e1c71b498 clean 2025-03-11 14:43:03 +01:00
harisreedhar ef313042c6 clean generator, discriminator and typing 2025-03-11 14:43:03 +01:00
harisreedhar a1fd382659 ugly training code 2025-03-11 14:43:03 +01:00
harisreedhar 650268c06b ugly training code 2025-03-11 14:43:03 +01:00
harisreedhar fe28b6fffe ugly training code 2025-03-11 14:43:02 +01:00
harisreedhar e6c2a64256 ugly training code 2025-03-11 14:43:02 +01:00
henryruhs 7bef17b551 Fix mypy 2025-03-11 14:43:02 +01:00
harisreedhar 8e53c6bc9f new swapper 2025-03-11 14:43:02 +01:00
henryruhs a461d9c389 Remove shebang 2025-01-15 09:28:00 +01:00
henryruhs f3e1c3feaa Improve naming for arcface converter trainer 2025-01-15 09:26:21 +01:00
Henry Ruhs abce63007d Update FUNDING.yml 2024-12-29 01:08:19 +01:00
henryruhs 5fa53dabf2 Move license to model directory 2024-12-06 11:55:40 +01:00
Henry Ruhs bd4725b52f Update dependencies 2024-11-11 21:54:04 +01:00
Henry Ruhs ef38a0e23f Modernize .gitignore 2024-10-30 08:21:30 +01:00