improve fallback mask

coder168 · Jul 8, 2023 · 2c4dd17 · 2c4dd17
1 parent c558363
commit 2c4dd17
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/inferencers/bisenet_mask_generator.py b/scripts/inferencers/bisenet_mask_generator.py
@@ -59,7 +59,7 @@ def generate_mask(
             mask = cv2.resize(mask, dsize=(w, h))
 
         if MaskGenerator.calculate_mask_coverage(mask) < fallback_ratio:
-            mask = self.fallback_mask_generator.generate_mask(face_image, face_area_on_image, use_minimal_area)
+            mask = self.fallback_mask_generator.generate_mask(face_image, face_area_on_image, use_minimal_area=True)
 
         return mask