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

feat: Masking only the top k largest predictions (#264)

Browse Source
This commit is contained in:
JudahGazit
2023-08-08 14:15:37 +03:00
committed by GitHub
parent 3e80b2d824
commit a1b498ff39
4 changed files with 25 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
scripts/!adetailer.py
View File

@@ -26,7 +26,7 @@ from adetailer import (
)
from adetailer.args import ALL_ARGS, BBOX_SORTBY, ADetailerArgs, EnableChecker
from adetailer.common import PredictOutput
from adetailer.mask import filter_by_ratio, mask_preprocess, sort_bboxes
from adetailer.mask import filter_take_largest, filter_by_ratio, mask_preprocess, sort_bboxes
from adetailer.traceback import rich_traceback
from adetailer.ui import adui, ordinal, suffix
from controlnet_ext import ControlNetExt, controlnet_exists, get_cn_models
@@ -463,6 +463,7 @@ class AfterDetailerScript(scripts.Script):
pred = filter_by_ratio(
pred, low=args.ad_mask_min_ratio, high=args.ad_mask_max_ratio
)
pred = filter_take_largest(pred, k=args.ad_mask_k_largest)
pred = self.sort_bboxes(pred)
return mask_preprocess(
pred.masks,