Fix: classification result 수정

2024-09-27 15:46:02 +09:00 · 2024-09-27 15:46:02 +09:00 · 669562b712
commit 669562b712
parent 3affff85f0
1 changed files with 1 additions and 1 deletions
--- a/ai/app/api/yolo/classfication.py
+++ b/ai/app/api/yolo/classfication.py
@ -118,7 +118,7 @@ async def classification_train(request: TrainRequest):
        recall= 0,
        mAP50= 0,
        mAP5095= 0,
-        accuracy=result["accuracy_top1"],
+        accuracy=result["metrics/accuracy_top1"],
        fitness= result["fitness"]
    )