vitordigitizing commited on
Commit
aeb16ab
·
verified ·
1 Parent(s): 8dce1e4

Update faceless_processing.py

Browse files
Files changed (1) hide show
  1. faceless_processing.py +1 -1
faceless_processing.py CHANGED
@@ -236,7 +236,7 @@ def build_interior_edges(rgb: np.ndarray, subject_mask: np.ndarray, cfg: Lineart
236
  erode_size = max(5, cfg.silhouette_line_thickness * 2 + 3)
237
  if erode_size % 2 == 0:
238
  erode_size += 1
239
- kernel_erode = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (erode_size, eroded_mask_size := erode_size))
240
  eroded_mask = cv2.erode(subject_mask, kernel_erode, iterations=1)
241
  edges = cv2.bitwise_and(edges, eroded_mask)
242
 
 
236
  erode_size = max(5, cfg.silhouette_line_thickness * 2 + 3)
237
  if erode_size % 2 == 0:
238
  erode_size += 1
239
+ kernel_erode = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (erode_size, erode_size))
240
  eroded_mask = cv2.erode(subject_mask, kernel_erode, iterations=1)
241
  edges = cv2.bitwise_and(edges, eroded_mask)
242