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

feat: add yolo world model

Browse Source
This commit is contained in:
Dowon
2024-03-01 10:54:19 +09:00
parent 692b052cfb
commit 9e9dcd5bca
7 changed files with 38 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
scripts/!adetailer.py
View File

@@ -695,6 +695,7 @@ class AfterDetailerScript(scripts.Script):
predictor = ultralytics_predict
ad_model = self.get_ad_model(args.ad_model)
kwargs["device"] = self.ultralytics_device
kwargs["classes"] = args.ad_model_classes
with change_torch_load():
pred = predictor(ad_model, pp.image, args.ad_confidence, **kwargs)