mirrors/facefusion
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Local model support for deep swapper part3

Browse Source
This commit is contained in:
henryruhs 2024-12-10 16:45:32 +01:00
parent 640d1c430e
commit 1b59f8ab73
2 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
facefusion/processors/choices.py
View File

@ -157,9 +157,11 @@ deep_swapper_models : List[DeepSwapperModel] =\
model_files = list_directory(resolve_relative_path('../.assets/models/local'))
for model_file in model_files:
model_id = '/'.join([ 'local', model_file.get('name') ])
deep_swapper_models.append(model_id)
if model_files:
for model_file in model_files:
model_id = '/'.join([ 'local', model_file.get('name') ])
deep_swapper_models.append(model_id)
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' ]

8
facefusion/processors/modules/deep_swapper.py
View File

@ -225,10 +225,10 @@ def create_static_model_set(download_scope : DownloadScope) -> ModelSet:
{
'sources':
{
'deep_swapper':
{
'path': resolve_relative_path(model_file.get('path'))
}
'deep_swapper':
{
'path': resolve_relative_path(model_file.get('path'))
}
},
'template': 'dfl_whole_face'
}