Bing-su/adetailer
1
0
Fork 0
mirror of https://github.com/Bing-su/adetailer.git synced 2026-04-25 16:59:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge remote-tracking branch 'origin/main' into dev

Browse Source
This commit is contained in:
Dowon
2024-09-25 20:13:04 +09:00
parent 753862017a 9ceb58685a
commit 9de46de1d2
7 changed files with 56 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
scripts/!adetailer.py
View File

@@ -52,7 +52,7 @@ from adetailer.args import (
from adetailer.common import PredictOutput, ensure_pil_image, safe_mkdir
from adetailer.mask import (
filter_by_ratio,
filter_k_largest,
filter_k_by,
has_intersection,
is_all_black,
mask_preprocess,
@@ -608,7 +608,7 @@ class AfterDetailerScript(scripts.Script):
pred = filter_by_ratio(
pred, low=args.ad_mask_min_ratio, high=args.ad_mask_max_ratio
)
pred = filter_k_largest(pred, k=args.ad_mask_k_largest)
pred = filter_k_by(pred, k=args.ad_mask_k, by=args.ad_mask_filter_method)
pred = self.sort_bboxes(pred)
masks = mask_preprocess(
pred.masks,