Spaces:

yemce
/

hair_filter

Running

yemce commited on Nov 7, 2024

Commit

bd60de7

verified ·

1 Parent(s): de82ff3

class 2 hair

Files changed (1) hide show

app.py CHANGED Viewed

@@ -41,7 +41,7 @@ def apply_filter(image, selected_color):
     pred_seg = upsampled_logits.argmax(dim=1)[0].numpy()
-    hair_mask = (pred_seg == 4).astype(np.uint8)
     # Orijinal görüntüyü OpenCV formatına dönüştürün ve HSV uzayına çevirin
     image_cv = cv.cvtColor(np.array(image), cv.COLOR_RGB2BGR)

     pred_seg = upsampled_logits.argmax(dim=1)[0].numpy()
+    hair_mask = (pred_seg == 2).astype(np.uint8)
     # Orijinal görüntüyü OpenCV formatına dönüştürün ve HSV uzayına çevirin
     image_cv = cv.cvtColor(np.array(image), cv.COLOR_RGB2BGR)