From 1c00d156f2e33fd07d8ab3c6d7e5fbe52a7c53e8 Mon Sep 17 00:00:00 2001
From: NNNNAI <844294823@qq.com>
Date: Mon, 19 Jul 2021 15:28:25 +0800
Subject: [PATCH] Update reverse2original.py

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

diff --git a/util/reverse2original.py b/util/reverse2original.py
index 25f06b5..86652e8 100644
--- a/util/reverse2original.py
+++ b/util/reverse2original.py
@@ -130,15 +130,15 @@ def reverse2wholeimage(b_align_crop_tenor_list,swaped_imgs, mats, crop_size, ori
 
         img_mask = img_white
 
-        if use_mask:
-            kernel = np.ones((10,10),np.uint8)
-            img_mask = cv2.erode(img_mask,kernel,iterations = 1)
-        else:
-            kernel = np.ones((40,40),np.uint8)
-            img_mask = cv2.erode(img_mask,kernel,iterations = 1)
-            kernel_size = (20, 20)
-            blur_size = tuple(2*i+1 for i in kernel_size)
-            img_mask = cv2.GaussianBlur(img_mask, blur_size, 0)
+        # if use_mask:
+        #     kernel = np.ones((40,40),np.uint8)
+        #     img_mask = cv2.erode(img_mask,kernel,iterations = 1)
+        # else:
+        kernel = np.ones((40,40),np.uint8)
+        img_mask = cv2.erode(img_mask,kernel,iterations = 1)
+        kernel_size = (20, 20)
+        blur_size = tuple(2*i+1 for i in kernel_size)
+        img_mask = cv2.GaussianBlur(img_mask, blur_size, 0)
 
         # kernel = np.ones((10,10),np.uint8)
         # img_mask = cv2.erode(img_mask,kernel,iterations = 1)