Bing-su/adetailer
1
0
Fork 0
mirror of https://github.com/Bing-su/adetailer.git synced 2026-02-21 15:44:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: boxes attribute error

Browse Source
This commit is contained in:
Bingsu
2023-04-26 23:08:30 +09:00
parent 44160f2b27
commit ff4f99ecf4
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

@@ -18,7 +18,7 @@ def ultralytics_predict(
model = YOLO(model_path)
pred = model(image, conf=confidence, hide_labels=True)
bboxes = pred[0].xyxy.cpu().numpy()
bboxes = pred[0].boxes.xyxy.cpu().numpy()
if bboxes.size == 0:
return PredictOutput()
bboxes = bboxes.tolist()