NNNNAI 4 лет назад
Родитель
Сommit
1faf9fd496
1 измененных файлов с 6 добавлено и 3 удалено
  1. 6 3
      util/reverse2original.py

+ 6 - 3
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)
+