From 0d4000b8799773d5072ddf30d6f98165d5f7f93f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=EA=B9=80=EC=A7=84=ED=98=84?= Date: Fri, 27 Sep 2024 11:31:40 +0900 Subject: [PATCH] =?UTF-8?q?Fix:=20download=5Fdata=20paremeter=20=EC=88=98?= =?UTF-8?q?=EC=A0=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ai/app/api/yolo/detection.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ai/app/api/yolo/detection.py b/ai/app/api/yolo/detection.py index 2517220..eb3f974 100644 --- 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)