Quellcode durchsuchen

Update reverse2original.py

NNNNAI vor 3 Jahren
Ursprung
Commit
1c00d156f2
1 geänderte Dateien mit 9 neuen und 9 gelöschten Zeilen
  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
 
-        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)