From 8335f6d15dbd4717222ad6302bef73ee614b9166 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=EA=B9=80=EC=A7=84=ED=98=84?= Date: Thu, 5 Sep 2024 10:36:17 +0900 Subject: [PATCH] =?UTF-8?q?Fix:=20Detection=20=EB=A0=88=EC=9D=B4=EB=B8=94?= =?UTF-8?q?=20=EA=B0=9D=EC=B2=B4=20=ED=8C=8C=EC=8B=B1=EC=97=90=EC=84=9C=20?= =?UTF-8?q?imageDepth=20=EC=88=98=EC=A0=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ai/app/api/yolo/detection.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/ai/app/api/yolo/detection.py b/ai/app/api/yolo/detection.py index 2d01a9f..b67914b 100644 --- a/ai/app/api/yolo/detection.py +++ b/ai/app/api/yolo/detection.py @@ -61,9 +61,9 @@ def predict(request: PredictRequest): for summary in result.summary() ], "split": "none", - "imageHeight": result.orig_shape[0], - "imageWidth": result.orig_shape[1], - "imageDepth": 1 + "imageHeight": result.orig_img.shape[0], + "imageWidth": result.orig_img.shape[1], + "imageDepth": result.orig_img.shape[2] } response.append({ "image_id":image.image_id,