Fix: Detection 레이블 객체 파싱에서 imageDepth 수정

2024-09-05 10:36:17 +09:00 · 2024-09-05 10:36:17 +09:00 · 8335f6d15d
commit 8335f6d15d
parent 635c2a2033
1 changed files with 3 additions and 3 deletions
--- 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,