From e7496dad94493bc8068ade3ff466cc9e06d2b678 Mon Sep 17 00:00:00 2001
From: NNNNAI <844294823@qq.com>
Date: Fri, 2 Jul 2021 19:16:33 +0800
Subject: [PATCH] Script for image face swapping(You can choose to add or not
 add a watermark )

---
 test_wholeimage_swapmutil.py  | 3 ++-
 test_wholeimage_swapsingle.py | 2 +-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/test_wholeimage_swapmutil.py b/test_wholeimage_swapmutil.py
index 5f39642..89048ad 100644
--- a/test_wholeimage_swapmutil.py
+++ b/test_wholeimage_swapmutil.py
@@ -73,7 +73,8 @@ if __name__ == '__main__':
         swap_result = model(None, b_align_crop_tenor, latend_id, None, True)[0]
         swap_result_list.append(swap_result)
 
-    reverse2wholeimage(swap_result_list, b_mat_list, crop_size, img_b_whole, logoclass, os.path.join(opt.output_path, 'result_whole_swapmutil.jpg'))
+
+    reverse2wholeimage(swap_result_list, b_mat_list, crop_size, img_b_whole, logoclass, os.path.join(opt.output_path, 'result_whole_swapmutil.jpg'),opt.no_simswaplogo)
     print(' ')
 
     print('************ Done ! ************')
diff --git a/test_wholeimage_swapsingle.py b/test_wholeimage_swapsingle.py
index edf2816..0e3fb02 100644
--- a/test_wholeimage_swapsingle.py
+++ b/test_wholeimage_swapsingle.py
@@ -72,7 +72,7 @@ if __name__ == '__main__':
         swap_result = model(None, b_align_crop_tenor, latend_id, None, True)[0]
         swap_result_list.append(swap_result)
 
-    reverse2wholeimage(swap_result_list, b_mat_list, crop_size, img_b_whole, logoclass, os.path.join(opt.output_path, 'result_whole_swapsingle.jpg'))
+    reverse2wholeimage(swap_result_list, b_mat_list, crop_size, img_b_whole, logoclass, os.path.join(opt.output_path, 'result_whole_swapsingle.jpg'),opt.no_simswaplogo)
 
     print(' ')