From ce35be047856fbb15f641199d02a045f69426c46 Mon Sep 17 00:00:00 2001 From: JSheng Pan <36942973+JShengP@users.noreply.github.com> Date: Thu, 3 Apr 2025 00:26:57 +0800 Subject: [PATCH] YOLOE: Fix visual prompt usage in docs (#19981) Signed-off-by: JSheng Pan <36942973+JShengP@users.noreply.github.com> --- docs/en/models/yoloe.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/en/models/yoloe.md b/docs/en/models/yoloe.md index a062e8bfc9..cd76d336a2 100644 --- a/docs/en/models/yoloe.md +++ b/docs/en/models/yoloe.md @@ -216,7 +216,7 @@ Object detection is straightforward with the `predict` method, as illustrated be # Set visual prompt visuals = dict( - bboxes=np.array([221.52, 405.8, 344.98, 857.54]), + bboxes=np.array([[221.52, 405.8, 344.98, 857.54]]), cls=np.array([0]), )