From b63562abad560d966285a3a076c923c346feb5dc Mon Sep 17 00:00:00 2001 From: henryruhs Date: Wed, 5 Mar 2025 22:43:09 +0100 Subject: [PATCH] Simplify infer() --- face_swapper/src/inferencing.py | 9 ++------- 1 file changed, 2 insertions(+), 7 deletions(-) diff --git a/face_swapper/src/inferencing.py b/face_swapper/src/inferencing.py index d1112db..47ade09 100644 --- a/face_swapper/src/inferencing.py +++ b/face_swapper/src/inferencing.py @@ -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)