sun15684712291/aicheckv2
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

修改部分问题

Browse Source
This commit is contained in:
sunyg
2025-02-26 10:04:10 +08:00
parent baadf6c197
commit 1f5cecc5ed
5 changed files with 19 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
app/api/business/project_train_api.py
View File

@ -15,6 +15,7 @@ from app.db.db_session import get_db
from typing import List
from fastapi import APIRouter, Depends, Request, UploadFile, File, Form
from fastapi.responses import StreamingResponse
from fastapi.encoders import jsonable_encoder
from sqlalchemy.orm import Session
"""项目管理API"""
@ -155,7 +156,8 @@ def get_image_list(project_id: int, session: Session = Depends(get_db)):
:return:
"""
image_list = pimc.get_image_list(project_id, session)
return rc.response_success(data=image_list)
result = jsonable_encoder(image_list)
return rc.response_success(data=result)
@project.post("/save_img_label")
@ -170,7 +172,7 @@ def save_img_label(img_leafer_label: ProjectImgLeaferLabel, session: Session = D
return rc.response_success(msg="保存成功")
@project.get("/get_img_leafer/{imager_id}")
@project.get("/get_img_leafer/{image_id}")
def get_img_leafer(image_id: int, session: Session = Depends(get_db)):
"""
根据图片id查询图片的leafer信息
@ -179,7 +181,10 @@ def get_img_leafer(image_id: int, session: Session = Depends(get_db)):
:return:
"""
img_leafer_out = ps.get_img_leafer(image_id, session)
return rc.response_success(data=img_leafer_out.leafer)
if img_leafer_out is None:
return rc.response_success()
else:
return rc.response_success(data=img_leafer_out['leafer'])
@project.get("/run_train/{project_id}")
@ -201,6 +206,7 @@ async def run_train(project_id: int, session: Session = Depends(get_db)):
media_type="text/plain")
@project.get("/get_train_list/{project_id}")
def get_train_list(project_id: int, session: Session = Depends(get_db)):
"""
根据项目id，获取训练列表
@ -209,7 +215,8 @@ def get_train_list(project_id: int, session: Session = Depends(get_db)):
:return:
"""
train_list = ptnc.get_train_list(project_id, session)
return rc.response_success(data=train_list)
result = jsonable_encoder(train_list)
return rc.response_success(data=result)