Fix: download_data paremeter 수정

2024-09-27 11:31:40 +09:00 · 2024-09-27 11:31:40 +09:00 · 0d4000b879
commit 0d4000b879
parent 8730b54d2d
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)