Hussein El-Hadidy
commited on
Commit
·
4d2f608
1
Parent(s):
ba736d8
Fix
Browse files- __pycache__/main.cpython-313.pyc +0 -0
- app.py +2 -1
- main.py +2 -1
__pycache__/main.cpython-313.pyc
CHANGED
|
Binary files a/__pycache__/main.cpython-313.pyc and b/__pycache__/main.cpython-313.pyc differ
|
|
|
app.py
CHANGED
|
@@ -230,8 +230,9 @@ async def process_image(file: UploadFile = File(...)):
|
|
| 230 |
save=False
|
| 231 |
)
|
| 232 |
resultsKeypointsXY = results[0].keypoints.xy
|
|
|
|
| 233 |
|
| 234 |
except Exception as e:
|
| 235 |
raise HTTPException(status_code=500, detail=f"YOLO processing error: {str(e)}")
|
| 236 |
|
| 237 |
-
return JSONResponse(content={"message": "Image processed successfully", "KeypointsXY": resultsKeypointsXY})
|
|
|
|
| 230 |
save=False
|
| 231 |
)
|
| 232 |
resultsKeypointsXY = results[0].keypoints.xy
|
| 233 |
+
confidences = results[0].boxes.conf
|
| 234 |
|
| 235 |
except Exception as e:
|
| 236 |
raise HTTPException(status_code=500, detail=f"YOLO processing error: {str(e)}")
|
| 237 |
|
| 238 |
+
return JSONResponse(content={"message": "Image processed successfully", "KeypointsXY": resultsKeypointsXY.tolist(), "confidences": confidences.tolist()})
|
main.py
CHANGED
|
@@ -230,8 +230,9 @@ async def process_image(file: UploadFile = File(...)):
|
|
| 230 |
save=False
|
| 231 |
)
|
| 232 |
resultsKeypointsXY = results[0].keypoints.xy
|
|
|
|
| 233 |
|
| 234 |
except Exception as e:
|
| 235 |
raise HTTPException(status_code=500, detail=f"YOLO processing error: {str(e)}")
|
| 236 |
|
| 237 |
-
return JSONResponse(content={"message": "Image processed successfully", "KeypointsXY": resultsKeypointsXY})
|
|
|
|
| 230 |
save=False
|
| 231 |
)
|
| 232 |
resultsKeypointsXY = results[0].keypoints.xy
|
| 233 |
+
confidences = results[0].boxes.conf
|
| 234 |
|
| 235 |
except Exception as e:
|
| 236 |
raise HTTPException(status_code=500, detail=f"YOLO processing error: {str(e)}")
|
| 237 |
|
| 238 |
+
return JSONResponse(content={"message": "Image processed successfully", "KeypointsXY": resultsKeypointsXY.tolist(), "confidences": confidences.tolist()})
|