Merge branch 'ai/feat/classification' into 'ai/develop'

Fix: classification result 수정 See merge request s11-s-project/S11P21S002!224
2024-09-27 15:46:52 +09:00 · 2024-09-27 15:46:52 +09:00 · 8d89997369
commit 8d89997369
parent 7ccd3b57d7 669562b712
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"]
    )