From 1faf9fd496b54e58b9f7b95c85f6841ab0822c73 Mon Sep 17 00:00:00 2001
From: NNNNAI <844294823@qq.com>
Date: Fri, 2 Jul 2021 19:16:59 +0800
Subject: [PATCH] Update reverse2original.py

---
 util/reverse2original.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/util/reverse2original.py b/util/reverse2original.py
index 9a27ecb..0a2f99c 100644
--- a/util/reverse2original.py
+++ b/util/reverse2original.py
@@ -3,7 +3,7 @@ import numpy as np
 # import  time
 from util.add_watermark import  watermark_image
 
-def reverse2wholeimage(swaped_imgs, mats, crop_size, oriimg, logoclass, save_path = '',):
+def reverse2wholeimage(swaped_imgs, mats, crop_size, oriimg, logoclass, save_path = '', no_simswaplogo = False):
 
     target_image_list = []
     img_mask_list = []
@@ -46,10 +46,13 @@ def reverse2wholeimage(swaped_imgs, mats, crop_size, oriimg, logoclass, save_pat
     img = np.array(oriimg, dtype=np.float)
     for img_mask, target_image in zip(img_mask_list, target_image_list):
         img = img_mask * target_image + (1-img_mask) * img
-
-    final_img = logoclass.apply_frames(img.astype(np.uint8))
+        
+    final_img = img.astype(np.uint8)
+    if not no_simswaplogo:
+        final_img = logoclass.apply_frames(final_img)
     cv2.imwrite(save_path, final_img)
 
     # cv2.imwrite('E:\\lny\\SimSwap-main\\output\\img_div.jpg', img * 255)
     # cv2.imwrite('E:\\lny\\SimSwap-main\\output\\ori_img.jpg', oriimg)
+    
     
\ No newline at end of file