From 2a578cf2164bf3472d4417650a40255f52afbb57 Mon Sep 17 00:00:00 2001
From: Dowon <ks2515@naver.com>
Date: Mon, 30 Oct 2023 21:58:27 +0900
Subject: [PATCH] fix: k largest sort order

---
 adetailer/mask.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/adetailer/mask.py b/adetailer/mask.py
index 913ea7e..d2f3680 100644
--- 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