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]), )