优化训练过程

app/model/crud/project_detect_crud.py

@@ -5,7 +5,7 @@ from fastapi.encoders import jsonable_encoder
 
 from app.model.bussiness_model import ProjectDetect, ProjectDetectImg, ProjectDetectLog, ProjectDetectLogImg
 from app.model.schemas.project_detect_schemas import ProjectDetectOut, ProjectDetectPager, \
-    ProjectDetectImageOut, ProjectDetectImgPager, ProjectDetectLogOut, ProjectDetectLogImgOut
+    ProjectDetectImageOut, ProjectDetectImgPager, ProjectDetectLogOut, ProjectDetectLogImgOut, ProjectDetectList
 from app.db.page_util import get_pager
 
 
@@ -59,7 +59,7 @@ def get_detect_list(project_id: int, session: Session):
     :return:
     """
     query = session.query(ProjectDetect).filter_by(project_id=project_id).order_by(asc(ProjectDetect.id))
-    result = jsonable_encoder(query.all())
+    result = [ProjectDetectList.from_orm(detect).dict() for detect in query.all()]
     return result
 
 

app/model/schemas/project_detect_schemas.py

@@ -33,6 +33,14 @@ class ProjectDetectOut(BaseModel):
         }
 
 
+class ProjectDetectList(BaseModel):
+    id: Optional[int]
+    detect_name: Optional[str]
+
+    class Config:
+        orm_mode = True
+
+
 class ProjectDetectImgPager(BaseModel):
     detect_id: Optional[int] = Field(..., description="训练集合id")
     pagerNum: Optional[int] = Field(None, description="当前页码")