Bing-su/adetailer
1
0
Fork 0
mirror of https://github.com/Bing-su/adetailer.git synced 2026-03-10 07:50:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: k largest sort order

Browse Source
This commit is contained in:
Dowon
2023-10-30 21:58:27 +09:00
parent d3e57b68d5
commit 2a578cf216
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
adetailer/mask.py
View File

@@ -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