Add deep swapper models by Rumateus

This commit is contained in:
henryruhs 2024-11-18 23:38:55 +01:00
parent c18d7a4d4f
commit c3f58b2d0f
3 changed files with 84 additions and 3 deletions

View File

@ -117,7 +117,34 @@ deep_swapper_models : List[DeepSwapperModel] =\
'mats/mica_suarez_320',
'mats/shailene_woodley_224',
'mats/shraddha_kapoor_320',
'mats/yu_jimin_352'
'mats/yu_jimin_352',
'rumateus/alison_brie_224',
'rumateus/amber_heard_224',
'rumateus/angelina_jolie_224',
'rumateus/aubrey_plaza_224',
'rumateus/bridget_regan_224',
'rumateus/cobie_smulders_224',
'rumateus/deborah_woll_224',
'rumateus/dua_lipa_224',
'rumateus/emma_stone_224',
'rumateus/hailee_steinfeld_224',
'rumateus/hilary_duff_224',
'rumateus/jessica_alba_224',
'rumateus/jessica_biel_224',
'rumateus/john_cena_224',
'rumateus/kim_kardashian_224',
'rumateus/kristen_bell_224',
'rumateus/lucy_liu_224',
'rumateus/margot_robbie_224',
'rumateus/megan_fox_224',
'rumateus/meghan_markle_224',
'rumateus/millie_bobby_brown_224',
'rumateus/natalie_portman_224',
'rumateus/nicki_minaj_224',
'rumateus/olivia_wilde_224',
'rumateus/shay_mitchell_224',
'rumateus/sophie_turner_224',
'rumateus/taylor_swift_224'
]
expression_restorer_models : List[ExpressionRestorerModel] = [ 'live_portrait' ]
face_debugger_items : List[FaceDebuggerItem] = [ 'bounding-box', 'face-landmark-5', 'face-landmark-5/68', 'face-landmark-68', 'face-landmark-68/5', 'face-mask', 'face-detector-score', 'face-landmarker-score', 'age', 'gender', 'race' ]

View File

@ -138,7 +138,34 @@ def create_model_set() -> ModelSet:
('mats', 'mica_suarez_320', (320, 320)),
('mats', 'shailene_woodley_224', (224, 224)),
('mats', 'shraddha_kapoor_320', (320, 320)),
('mats', 'yu_jimin_352', (352, 352))
('mats', 'yu_jimin_352', (352, 352)),
('rumateus', 'alison_brie_224', (224, 224)),
('rumateus', 'amber_heard_224', (224, 224)),
('rumateus', 'angelina_jolie_224', (224, 224)),
('rumateus', 'aubrey_plaza_224', (224, 224)),
('rumateus', 'bridget_regan_224', (224, 224)),
('rumateus', 'cobie_smulders_224', (224, 224)),
('rumateus', 'deborah_woll_224', (224, 224)),
('rumateus', 'dua_lipa_224', (224, 224)),
('rumateus', 'emma_stone_224', (224, 224)),
('rumateus', 'hailee_steinfeld_224', (224, 224)),
('rumateus', 'hilary_duff_224', (224, 224)),
('rumateus', 'jessica_alba_224', (224, 224)),
('rumateus', 'jessica_biel_224', (224, 224)),
('rumateus', 'john_cena_224', (224, 224)),
('rumateus', 'kim_kardashian_224', (224, 224)),
('rumateus', 'kristen_bell_224', (224, 224)),
('rumateus', 'lucy_liu_224', (224, 224)),
('rumateus', 'margot_robbie_224', (224, 224)),
('rumateus', 'megan_fox_224', (224, 224)),
('rumateus', 'meghan_markle_224', (224, 224)),
('rumateus', 'millie_bobby_brown_224', (224, 224)),
('rumateus', 'natalie_portman_224', (224, 224)),
('rumateus', 'nicki_minaj_224', (224, 224)),
('rumateus', 'olivia_wilde_224', (224, 224)),
('rumateus', 'shay_mitchell_224', (224, 224)),
('rumateus', 'sophie_turner_224', (224, 224)),
('rumateus', 'taylor_swift__224', (224, 224))
]
model_set : ModelSet = {}

View File

@ -118,7 +118,34 @@ DeepSwapperModel = Literal\
'mats/mica_suarez_320',
'mats/shailene_woodley_224',
'mats/shraddha_kapoor_320',
'mats/yu_jimin_352'
'mats/yu_jimin_352',
'rumateus/alison_brie_224',
'rumateus/amber_heard_224',
'rumateus/angelina_jolie_224',
'rumateus/aubrey_plaza_224',
'rumateus/bridget_regan_224',
'rumateus/cobie_smulders_224',
'rumateus/deborah_woll_224',
'rumateus/dua_lipa_224',
'rumateus/emma_stone_224',
'rumateus/hailee_steinfeld_224',
'rumateus/hilary_duff_224',
'rumateus/jessica_alba_224',
'rumateus/jessica_biel_224',
'rumateus/john_cena_224',
'rumateus/kim_kardashian_224',
'rumateus/kristen_bell_224',
'rumateus/lucy_liu_224',
'rumateus/margot_robbie_224',
'rumateus/megan_fox_224',
'rumateus/meghan_markle_224',
'rumateus/millie_bobby_brown_224',
'rumateus/natalie_portman_224',
'rumateus/nicki_minaj_224',
'rumateus/olivia_wilde_224',
'rumateus/shay_mitchell_224',
'rumateus/sophie_turner_224',
'rumateus/taylor_swift_224'
]
ExpressionRestorerModel = Literal['live_portrait']
FaceDebuggerItem = Literal['bounding-box', 'face-landmark-5', 'face-landmark-5/68', 'face-landmark-68', 'face-landmark-68/5', 'face-mask', 'face-detector-score', 'face-landmarker-score', 'age', 'gender', 'race']