Update faceless_processing.py

faceless_processing.py

@@ -723,12 +723,38 @@ def _skeletonize(img: np.ndarray) -> np.ndarray:
 
 
 def _build_interior_edges_canny(rgb: np.ndarray, subject_mask: np.ndarray, cfg: LineartConfig) -> np.ndarray:
+    h, w = rgb.shape[:2]
+    max_dim = max(w, h)
+    scale = max_dim / 512.0
+
     gray = cv2.cvtColor(rgb, cv2.COLOR_RGB2GRAY)
-    blurred = cv2.bilateralFilter(gray, cfg.bilateral_d, cfg.bilateral_sigma_color, cfg.bilateral_sigma_space)
-    blurred = cv2.GaussianBlur(blurred, (cfg.gaussian_ksize, cfg.gaussian_ksize), 0)
+    
+    if getattr(cfg, 'is_cartoon', False):
+        # Para desenhos/cartoons, aplicamos um filtro bilateral muito mais forte para simplificar texturas (como pelos de animais ou texturas de pele)
+        d_bilateral = int(15 * scale)
+        if d_bilateral % 2 == 0:
+            d_bilateral += 1
+        d_bilateral = max(9, d_bilateral)
+        
+        blurred = cv2.bilateralFilter(gray, d_bilateral, 150, 150)
+        
+        k_gaussian = int(9 * scale)
+        if k_gaussian % 2 == 0:
+            k_gaussian += 1
+        k_gaussian = max(5, k_gaussian)
+        blurred = cv2.GaussianBlur(blurred, (k_gaussian, k_gaussian), 0)
+    else:
+        blurred = cv2.bilateralFilter(gray, cfg.bilateral_d, cfg.bilateral_sigma_color, cfg.bilateral_sigma_space)
+        blurred = cv2.GaussianBlur(blurred, (cfg.gaussian_ksize, cfg.gaussian_ksize), 0)
     
     if getattr(cfg, 'is_cartoon', False):
         # 1. Binarização adaptativa inteligente para extrair traços cheios (evita contorno duplo!)
+        # Usamos uma janela adaptativa maior para capturar linhas de estrutura e ignorar texturas locais
+        block_size = int(65 * scale)
+        if block_size % 2 == 0:
+            block_size += 1
+        block_size = max(25, block_size)
+        
         corners = [blurred[0, 0], blurred[0, -1], blurred[-1, 0], blurred[-1, -1]]
         bg = float(np.median(corners))
         if bg > 127: # Fundo claro, traços escuros
@@ -736,17 +762,17 @@ def _build_interior_edges_canny(rgb: np.ndarray, subject_mask: np.ndarray, cfg:
                 blurred, 255,
                 cv2.ADAPTIVE_THRESH_GAUSSIAN_C,
                 cv2.THRESH_BINARY_INV,
-                21, 5
+                block_size, 8
             )
         else: # Fundo escuro, traços claros
             thresh = cv2.adaptiveThreshold(
                 blurred, 255,
                 cv2.ADAPTIVE_THRESH_GAUSSIAN_C,
                 cv2.THRESH_BINARY,
-                21, 5
+                block_size, 8
             )
             
-        # 2. Canny complementar para não perder pequenos detalhes de alto contraste
+        # 2. Canny complementar para não perder pequenos detalhes de alto contraste (ex: olhos)
         canny_edges = cv2.Canny(blurred, cfg.canny_low, cfg.canny_high)
         # Dilata Canny levemente para fundir com os traços do limiar adaptativo
         kernel_c = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (3, 3))
@@ -769,12 +795,17 @@ def _build_interior_edges_canny(rgb: np.ndarray, subject_mask: np.ndarray, cfg:
 
     contours, _ = cv2.findContours(edges, cv2.RETR_LIST, cv2.CHAIN_APPROX_SIMPLE)
     
+    # Determinamos o perímetro mínimo de forma dinâmica para descartar pequenos riscos (ex: pelos)
+    min_perimeter = cfg.min_contour_perimeter
+    if getattr(cfg, 'is_cartoon', False):
+        min_perimeter = max(min_perimeter, 45.0 * scale)
+
     smoothed_contours = []
     for cnt in contours:
         perimeter = cv2.arcLength(cnt, False)
-        if perimeter < cfg.min_contour_perimeter:
+        if perimeter < min_perimeter:
             continue
-        eps = 0.0012 * perimeter
+        eps = 0.001 * perimeter if getattr(cfg, 'is_cartoon', False) else 0.0012 * perimeter
         smooth = cv2.approxPolyDP(cnt, eps, False)
         smoothed_contours.append(smooth)
         
@@ -785,7 +816,7 @@ def _build_interior_edges_canny(rgb: np.ndarray, subject_mask: np.ndarray, cfg:
         contours=smoothed_contours,
         min_dist_px=min_dist,
         overlap_threshold=overlap_thresh,
-        min_perimeter=cfg.min_contour_perimeter,
+        min_perimeter=min_perimeter,
         thickness=cfg.interior_line_thickness,
         shape=edges.shape
     )