补全内容

2025-02-21 14:28:44 +08:00
parent 85d0a8fadc
commit e74cb852c4
10 changed files with 207 additions and 13 deletions
--- a/app/api/business/project_api.py
+++ b/app/api/business/project_api.py
@ -159,4 +159,7 @@ def run_train(project_id: int, session: Session = Depends(get_db)):
        return rc.response_error("项目查询错误")
    if project_info.project_status == '1':
        return rc.response_error("项目当前存在训练进程，请稍后再试")
-    return StreamingResponse(ps.run_train_yolo(project_info, session), media_type="text/plain")
+    data, project, name, epochs, yolo_path, version_path = ps.run_train_yolo(project_info, session)
+    return StreamingResponse(
+        ps.run_commend(data, project, name, epochs, yolo_path, version_path, project_id, session),
+        media_type="text/plain")