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

style: ruff RET

Browse Source
This commit is contained in:
Bingsu
2023-06-22 11:23:58 +09:00
parent 6337bd2a99
commit 87c3cf3e4f
2 changed files with 2 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
adetailer/mask.py
View File

@@ -129,9 +129,7 @@ def mask_preprocess(
masks = [dilate_erode(m, kernel) for m in masks]
masks = [m for m in masks if not is_all_black(m)]
masks = mask_merge_invert(masks, mode=merge_invert)
return masks
return mask_merge_invert(masks, mode=merge_invert)
# Bbox sorting

3
scripts/!adetailer.py
View File

@@ -423,8 +423,7 @@ class AfterDetailerScript(scripts.Script):
def sort_bboxes(self, pred: PredictOutput) -> PredictOutput:
sortby = opts.data.get("ad_bbox_sortby", BBOX_SORTBY[0])
sortby_idx = BBOX_SORTBY.index(sortby)
pred = sort_bboxes(pred, sortby_idx)
return pred
return sort_bboxes(pred, sortby_idx)
def pred_preprocessing(self, pred: PredictOutput, args: ADetailerArgs):
pred = filter_by_ratio(