CalvinBackup/facefusion-labs
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion-labs.git synced 2026-07-28 16:08:50 +02:00
Code Issues Packages Projects Releases Wiki Activity

Simplify infer()

Browse Source
This commit is contained in:
henryruhs
2025-03-11 14:43:10 +01:00
parent e9ea9cd9e5
commit b63562abad
1 changed files with 2 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
face_swapper/src/inferencing.py
+2 -7
View File
@@ -12,12 +12,6 @@ CONFIG = configparser.ConfigParser()
CONFIG.read('config.ini')
def run_swap(generator : GeneratorModule, embedder : EmbedderModule, source_tensor : Tensor, target_tensor : Tensor) -> Tensor:
source_embedding = calc_embedding(embedder, source_tensor, (0, 0, 0, 0))
output_tensor = generator(source_embedding, target_tensor)[0]
return output_tensor
def infer() -> None:
generator_path = CONFIG.get('inferencing', 'generator_path')
embedder_path = CONFIG.get('inferencing', 'embedder_path')
@@ -34,5 +28,6 @@ def infer() -> None:
source_tensor = io.read_image(source_path)
target_tensor = io.read_image(target_path)
output_tensor = run_swap(generator, embedder, source_tensor, target_tensor)
source_embedding = calc_embedding(embedder, source_tensor, (0, 0, 0, 0))
output_tensor = generator(source_embedding, target_tensor)[0]
io.write_jpeg(output_tensor, output_path)