jhynsoo/worlabel
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
김진현 2024-09-05 10:36:17 +09:00
parent 635c2a2033
commit 8335f6d15d
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
ai/app/api/yolo/detection.py
View File

@ -61,9 +61,9 @@ def predict(request: PredictRequest):
for summary in result.summary() for summary in result.summary()
], ],
"split": "none", "split": "none",
"imageHeight": result.orig_shape[0], "imageHeight": result.orig_img.shape[0],
"imageWidth": result.orig_shape[1], "imageWidth": result.orig_img.shape[1],
"imageDepth": 1 "imageDepth": result.orig_img.shape[2]
} }
response.append({ response.append({
"image_id":image.image_id, "image_id":image.image_id,