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

Fix: download_data paremeter 수정 See merge request s11-s-project/S11P21S002!213
2024-09-27 11:32:32 +09:00 · 2024-09-27 11:32:32 +09:00 · 2d6fdd23ff
commit 2d6fdd23ff
parent 959a16b59c 0d4000b879
1 changed files with 1 additions and 1 deletions
--- a/ai/app/api/yolo/detection.py
+++ b/ai/app/api/yolo/detection.py
@ -123,7 +123,7 @@ async def detection_train(request: TrainRequest):
    train_data, val_data = split_data(request.data, request.ratio)

    # 데이터 전처리: 데이터 이미지 및 레이블 다운로드
-    download_data(train_data, val_data, dataset_root_path, request.label_map)
+    download_data(train_data, val_data, dataset_root_path)

    # 학습
    results = run_train(request, model,dataset_root_path)