From 5567b49a6d386f063466c5d7bf5259878b424eac Mon Sep 17 00:00:00 2001 From: harisreedhar Date: Tue, 11 Mar 2025 23:02:46 +0530 Subject: [PATCH] fix --- face_swapper/src/exporting.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/face_swapper/src/exporting.py b/face_swapper/src/exporting.py index 3241ad3..31fc078 100644 --- a/face_swapper/src/exporting.py +++ b/face_swapper/src/exporting.py @@ -14,7 +14,7 @@ def export() -> None: config_directory_path = CONFIG_PARSER.get('exporting', 'directory_path') config_source_path = CONFIG_PARSER.get('exporting', 'source_path') config_target_path = CONFIG_PARSER.get('exporting', 'target_path') - config_output_size = CONFIG_PARSER.getint('training.model.generator', 'output_size') + config_target_size = CONFIG_PARSER.getint('exporting', 'target_size') config_ir_version = CONFIG_PARSER.getint('exporting', 'ir_version') config_opset_version = CONFIG_PARSER.getint('exporting', 'opset_version') @@ -22,5 +22,5 @@ def export() -> None: model = FaceSwapperTrainer.load_from_checkpoint(config_source_path, config_parser = CONFIG_PARSER, map_location = 'cpu').eval() model.ir_version = torch.tensor(config_ir_version) source_tensor = torch.randn(1, 512) - target_tensor = torch.randn(1, 3, config_output_size, config_output_size) + target_tensor = torch.randn(1, 3, config_target_size, config_target_size) torch.onnx.export(model, (source_tensor, target_tensor), config_target_path, input_names = [ 'source', 'target' ], output_names = [ 'output', 'mask' ], opset_version = config_opset_version)