From 1fea38471de281cd80c33c3a3e23b31afbf07357 Mon Sep 17 00:00:00 2001 From: Bingsu Date: Wed, 26 Apr 2023 23:18:57 +0900 Subject: [PATCH] =?UTF-8?q?fix:=20ultralytics=20hide=5Flabels=20=E2=86=92?= =?UTF-8?q?=20show=5Flabels?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- adetailer/ultralytics.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/adetailer/ultralytics.py b/adetailer/ultralytics.py index 927de7a..8629051 100644 --- a/adetailer/ultralytics.py +++ b/adetailer/ultralytics.py @@ -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: