[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
Bing-su · Sep 17, 2024 · fb96474 · fb96474
1 parent 0850ba5
commit fb96474
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 5 deletions.
diff --git a/aaaaaa/ui.py b/aaaaaa/ui.py
@@ -295,11 +295,11 @@ def detection(w: Widgets, n: int, is_img2img: bool):
                 elem_id=eid("ad_confidence"),
             )
             w.ad_mask_filter_method = gr.Radio(
-                choices=["Area", "Confidence"], 
+                choices=["Area", "Confidence"],
                 value="Area",
                 label="Method to filter top k masks by (confidence or area)",
                 visible=True,
-                elem_id=eid("ad_mask_filter_method")
+                elem_id=eid("ad_mask_filter_method"),
             )
             w.ad_mask_k = gr.Slider(
                 label="Mask only the top k (0 to disable)" + suffix(n),

diff --git a/adetailer/mask.py b/adetailer/mask.py
@@ -238,6 +238,7 @@ def filter_k_largest(pred: PredictOutput[T], k: int = 0) -> PredictOutput[T]:
     pred.masks = [pred.masks[i] for i in idx]
     return pred
 
+
 def filter_k_most_confident(pred: PredictOutput[T], k: int = 0) -> PredictOutput[T]:
     if not pred.bboxes or not pred.confidences or k == 0:
         return pred
@@ -247,6 +248,7 @@ def filter_k_most_confident(pred: PredictOutput[T], k: int = 0) -> PredictOutput
     pred.masks = [pred.masks[i] for i in idx]
     return pred
 
+
 # Merge / Invert
 def mask_merge(masks: list[Image.Image]) -> list[Image.Image]:
     arrs = [np.array(m) for m in masks]

diff --git a/adetailer/mediapipe.py b/adetailer/mediapipe.py
@@ -70,7 +70,9 @@ def mediapipe_face_detection(
     masks = create_mask_from_bbox(bboxes, image.size)
     preview = Image.fromarray(preview_array)
 
-    return PredictOutput(bboxes=bboxes, masks=masks, confidences=confidences, preview=preview)
+    return PredictOutput(
+        bboxes=bboxes, masks=masks, confidences=confidences, preview=preview
+    )
 
 
 def mediapipe_face_mesh(
@@ -157,7 +159,7 @@ def mediapipe_face_mesh_eyes_only(
             for outline in (left_outline, right_outline):
                 draw.polygon(outline, fill="white")
             masks.append(mask)
-            
+
         bboxes = create_bbox_from_mask(masks, image.size)
         preview = draw_preview(preview, bboxes, masks)
         return PredictOutput(bboxes=bboxes, masks=masks, preview=preview)

diff --git a/adetailer/ultralytics.py b/adetailer/ultralytics.py
@@ -44,7 +44,9 @@ def ultralytics_predict(
     preview = cv2.cvtColor(preview, cv2.COLOR_BGR2RGB)
     preview = Image.fromarray(preview)
 
-    return PredictOutput(bboxes=bboxes, masks=masks, confidences=confidences, preview=preview)
+    return PredictOutput(
+        bboxes=bboxes, masks=masks, confidences=confidences, preview=preview
+    )
 
 
 def apply_classes(model: YOLO | YOLOWorld, model_path: str | Path, classes: str):