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

Fix: get_model parameter 수정 See merge request s11-s-project/S11P21S002!212
2024-09-27 11:29:37 +09:00 · 2024-09-27 11:29:37 +09:00 · 959a16b59c
commit 959a16b59c
parent a93825ea8f 8730b54d2d
1 changed files with 1 additions and 1 deletions
--- a/ai/app/api/yolo/detection.py
+++ b/ai/app/api/yolo/detection.py
@ -111,7 +111,7 @@ async def detection_train(request: TrainRequest):
    dataset_root_path = get_dataset_root_path(request.project_id)

    # 모델 로드
-    model = get_model(request)
+    model = get_model(request.project_id, request.m_key)

    # 이 값을 학습할때 넣으면 이 카테고리들이 학습됨
    names = list(request.label_map)