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

fix: ultralytics hide_labels → show_labels

Browse Source
This commit is contained in:
Bingsu
2023-04-26 23:18:57 +09:00
parent ff4f99ecf4
commit 1fea38471d
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

@@ -16,7 +16,7 @@ def ultralytics_predict(
model_path = str(model_path)
model = YOLO(model_path)
pred = model(image, conf=confidence, hide_labels=True)
pred = model(image, conf=confidence, show_labels=False)
bboxes = pred[0].boxes.xyxy.cpu().numpy()
if bboxes.size == 0: