Bing-su/adetailer
1
0
Fork 0
mirror of https://github.com/Bing-su/adetailer.git synced 2026-05-01 03:31:21 +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 = [dilate_erode(m, kernel) for m in masks]
masks = [m for m in masks if not is_all_black(m)] masks = [m for m in masks if not is_all_black(m)]
masks = mask_merge_invert(masks, mode=merge_invert) return mask_merge_invert(masks, mode=merge_invert)
return masks
# Bbox sorting # Bbox sorting

3
scripts/!adetailer.py
View File

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