fix: k largest sort order

Kuron88 · Oct 30, 2023 · 2a578cf · 2a578cf
1 parent d3e57b6
commit 2a578cf
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/adetailer/mask.py b/adetailer/mask.py
@@ -220,6 +220,7 @@ def filter_k_largest(pred: PredictOutput, k: int = 0) -> PredictOutput:
         return pred
     areas = [bbox_area(bbox) for bbox in pred.bboxes]
     idx = np.argsort(areas)[-k:]
+    idx = idx[::-1]
     pred.bboxes = [pred.bboxes[i] for i in idx]
     pred.masks = [pred.masks[i] for i in idx]
     return pred