浏览代码

Update reverse2original.py

NNNNAI 3 年之前
父节点
当前提交
1c00d156f2
共有 1 个文件被更改,包括 9 次插入9 次删除
  1. 9 9
      util/reverse2original.py

+ 9 - 9
util/reverse2original.py

@@ -130,15 +130,15 @@ def reverse2wholeimage(b_align_crop_tenor_list,swaped_imgs, mats, crop_size, ori
 
 
         img_mask = img_white
         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)
         # kernel = np.ones((10,10),np.uint8)
         # img_mask = cv2.erode(img_mask,kernel,iterations = 1)
         # img_mask = cv2.erode(img_mask,kernel,iterations = 1)