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-28 16:30:11 +08:00
parent 0301e41e96
commit 4262d3e908
20 changed files with 564 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

48
app/model/bussiness_model.py
View File

@ -27,6 +27,7 @@ class ProjectInfo(DbCommon):
project_status: Mapped[str] = mapped_column(String(10))
user_id: Mapped[int] = mapped_column(Integer)
train_version: Mapped[int] = mapped_column(Integer)
del_flag: Mapped[int] = mapped_column(Integer)
class ProjectLabel(DbCommon):
@ -73,7 +74,9 @@ class ProjectImgLabel(DbCommon):
class ProjectTrain(DbCommon):
"""项目训练版本信息表"""
"""
项目训练版本信息表
"""
__tablename__ = "project_train"
project_id: Mapped[int] = mapped_column(Integer, nullable=False)
train_version: Mapped[str] = mapped_column(String(32), nullable=False)
@ -83,9 +86,48 @@ class ProjectTrain(DbCommon):
class ProjectDetect(DbCommon):
"""
训练推理集合
项目推理集合
"""
__tablename__ = "project_detect"
project_id: Mapped[str] = mapped_column(Integer, nullable=False)
project_id: Mapped[int] = mapped_column(Integer, nullable=False)
detect_name: Mapped[str] = mapped_column(String(64), nullable=False)
detect_version: Mapped[int] = mapped_column(Integer)
detect_no: Mapped[str] = mapped_column(String(32))
detect_status: Mapped[int] = mapped_column(Integer)
file_type: Mapped[str] = mapped_column(String(10))
folder_url: Mapped[str] = mapped_column(String(255))
class ProjectDetectImg(DbCommon):
"""
推理之前的图片
"""
__tablename__ = "project_detect_img"
detect_id: Mapped[int] = mapped_column(Integer, nullable=False)
file_name: Mapped[str] = mapped_column(String(64), nullable=False)
image_url: Mapped[str] = mapped_column(String(255), nullable=False)
thumb_image_url: Mapped[str] = mapped_column(String(255), nullable=False)
class ProjectDetectLog(DbCommon):
"""
项目推理记录
"""
__tablename__ = "project_detect_log"
detect_id: Mapped[int] = mapped_column(Integer, nullable=False)
detect_version: Mapped[str] = mapped_column(String(10))
train_id: Mapped[int] = mapped_column(Integer, nullable=False)
train_version: Mapped[str] = mapped_column(String(10))
pt_type: Mapped[str] = mapped_column(String(32))
folder_url: Mapped[str] = mapped_column(String(255))
detect_folder_url: Mapped[str] = mapped_column(String(255))
class ProjectDetectLogImg(DbCommon):
"""
推理完成的图片
"""
__tablename__ = "project_detect_log_img"
log_id: Mapped[int] = mapped_column(Integer, nullable=False)
file_name: Mapped[str] = mapped_column(String(64), nullable=False)
image_url: Mapped[str] = mapped_column(String(255), nullable=False)