first commit

app/schemas/TrainResult.py

@@ -0,0 +1,64 @@
+"""
+@Time ： 2022/9/29 11:39
+@Auth ： 东
+@File ：TrainResult.py
+@IDE ：PyCharm
+@Motto：ABC(Always Be Coding)
+@Desc：训练报告结果类
+
+"""
+import datetime
+from typing import List, Dict
+
+from pydantic import BaseModel, Field
+
+
+class ProcessValueList(BaseModel):
+    name: str = Field(..., description='名称')
+    value: List[int] = Field(..., description='过程值，如损失值，精度等')
+
+
+class Report(BaseModel):
+    """
+    训练算法返回值规范
+    """
+    id: str = Field(..., description='唯一值')
+    rate_of_progess: float = Field(..., description='进度，保留一位小数')
+    precision: List[ProcessValueList] = Field(..., description="过程值列表")
+    sum: int = Field(..., description='总轮次')
+    progress: int = Field(..., description='当前轮次')
+    isfinish: int = Field(0, description="是否结束")
+    num_train_img: int = Field(..., description="参与训练图像数量")
+    train_mod_savepath: str = Field(..., description="模型保存路径")
+    start_time: datetime.date = Field(datetime.datetime.now(), description="开始时间")
+    end_time: datetime.date = Field(datetime.datetime.now(), description="结束时间")
+
+
+class ReportDict(BaseModel):
+    """
+    验证算法返回值规范
+    """
+    id: str = Field(..., description='唯一值')
+    rate_of_progess: float = Field(..., description='进度，保留一位小数')
+    precision: List[Dict] = Field(..., description="过程值列表")
+    start_time: datetime.date = Field(datetime.datetime.now(), description="开始时间")
+    end_time: datetime.date = Field(datetime.datetime.now(), description="结束时间")
+
+
+class DetectProcessValueDice(BaseModel):
+    """
+    检测算法中间值
+    """
+    ori_img: str = Field(..., description='原时图片路径， 绝对路径')
+    res_img: str = Field(..., description='结果图片路径， 绝对路径')
+
+
+class DetectReport(BaseModel):
+    """
+    检测算法返回值规范
+    """
+    id: str = Field(..., description='唯一值')
+    rate_of_progess: float = Field(..., description='进度，保留一位小数')
+    precision: List[DetectProcessValueDice] = Field(..., description="过程值列表")
+    start_time: datetime.date = Field(datetime.datetime.now(), description="开始时间")
+    end_time: datetime.date = Field(datetime.datetime.now(), description="结束时间")

app/schemas/__init__.py

@@ -0,0 +1,5 @@
+from app.core.common_utils import import_subs
+
+__all__ = import_subs(locals())
+
+