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

fix: sync with ultralytics implement

Browse Source 
f8e681c2be/ultralytics/models/yolo/model.py (L17)
This commit is contained in:
Dowon
2024-02-29 20:34:35 +09:00
parent 6dcad30b64
commit 6018184ce4
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
adetailer/ultralytics.py
View File

@@ -40,7 +40,7 @@ def ultralytics_predict(
def apply_classes(model, model_path: str | Path, classes: str): def apply_classes(model, model_path: str | Path, classes: str):
if not classes or not Path(model_path).stem.endswith("world"): if not classes or "-world" not in Path(model_path).stem:
return return
parsed = [c.strip() for c in classes.split(",")] parsed = [c.strip() for c in classes.split(",")]
model.set_classes(parsed) model.set_classes(parsed)