完善推理模块的主体功能

This commit is contained in:
2025-03-05 16:16:12 +08:00
parent 5f4ed473d7
commit 05e2e8a59a
3 changed files with 5 additions and 5 deletions

View File

@ -34,8 +34,8 @@ def add_detect(detect_in: ProjectDetectIn, session: Session = Depends(get_db)):
:param session:
:return:
"""
pds.add_detect(detect_in, session)
return rc.response_success("新增成功")
detect = pds.add_detect(detect_in, session)
return rc.response_success(msg="新增成功", data=detect.id)
@detect.post("/get_img_list")
@ -89,7 +89,7 @@ def del_detect_img(detect_img_id: int, session: Session = Depends(get_db)):
return rc.response_error(msg="删除失败")
@detect.get("/run_detect_yolo")
@detect.post("/run_detect_yolo")
def run_detect_yolo(detect_log_in: ProjectDetectLogIn, session: Session = Depends(get_db)):
"""
开始执行训练

View File

@ -14,7 +14,7 @@ from app.common.redis_cli import redis_conn
login = APIRouter()
@login.post("/")
@login.post("")
def login_check(user: SysUserLogin, session: Session = Depends(get_db)):
"""
登录验证并返回token