143 lines
5.4 KiB
Python
143 lines
5.4 KiB
Python
from app.db.db_base import DbCommon
|
||
from sqlalchemy import String, Integer, JSON
|
||
from sqlalchemy.orm import Mapped, mapped_column
|
||
|
||
|
||
class ProjectType(DbCommon):
|
||
"""
|
||
项目类别表 - 标识项目的类型目前存在的（目标识别，OCR识别，瑕疵检测，图像分类）
|
||
"""
|
||
__tablename__ = "project_type"
|
||
type_code: Mapped[str] = mapped_column(String(20), unique=True, nullable=False)
|
||
type_name: Mapped[str] = mapped_column(String(20))
|
||
icon_path: Mapped[str] = mapped_column(String(255))
|
||
description: Mapped[str] = mapped_column(String(255))
|
||
type_status: Mapped[str] = mapped_column(String(10))
|
||
|
||
|
||
class ProjectInfo(DbCommon):
|
||
"""
|
||
项目信息表
|
||
"""
|
||
__tablename__ = "project_info"
|
||
project_no: Mapped[str] = mapped_column(String(32), unique=True, nullable=False)
|
||
project_name: Mapped[str] = mapped_column(String(32), unique=True, nullable=False)
|
||
type_code: Mapped[str] = mapped_column(String(10))
|
||
description: Mapped[str] = mapped_column(String(255))
|
||
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):
|
||
"""
|
||
项目标签表
|
||
"""
|
||
__tablename__ = "project_label"
|
||
label_name: Mapped[str] = mapped_column(String(32), unique=True, nullable=False)
|
||
project_id: Mapped[int] = mapped_column(Integer, nullable=False)
|
||
meta: Mapped[dict] = mapped_column(JSON)
|
||
|
||
|
||
class ProjectImage(DbCommon):
|
||
"""
|
||
项目图片表
|
||
"""
|
||
__tablename__ = "project_image"
|
||
img_type: Mapped[str] = mapped_column(String(10))
|
||
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)
|
||
project_id: Mapped[int] = mapped_column(Integer)
|
||
|
||
|
||
class ProjectImgLeafer(DbCommon):
|
||
"""
|
||
项目图片leafer表
|
||
"""
|
||
__tablename__ = "project_img_leafer"
|
||
image_id: Mapped[int] = mapped_column(Integer, nullable=False)
|
||
leafer: Mapped[dict] = mapped_column(JSON)
|
||
|
||
|
||
class ProjectImgLabel(DbCommon):
|
||
"""
|
||
项目图片标签对应表，一张图片对应多个label
|
||
"""
|
||
__tablename__ = "project_img_label"
|
||
image_id: Mapped[int] = mapped_column(Integer, nullable=False)
|
||
label_id: Mapped[int] = mapped_column(Integer, nullable=False)
|
||
mark_center_x: Mapped[str] = mapped_column(String(64), nullable=False)
|
||
mark_center_y: Mapped[str] = mapped_column(String(64), nullable=False)
|
||
mark_width: Mapped[str] = mapped_column(String(64), nullable=False)
|
||
mark_height: Mapped[str] = mapped_column(String(64), nullable=False)
|
||
|
||
|
||
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)
|
||
train_url: Mapped[str] = mapped_column(String(255), nullable=False)
|
||
train_data: Mapped[str] = mapped_column(String(255), nullable=False)
|
||
weights_id: Mapped[int] = mapped_column(Integer)
|
||
weights_name: Mapped[str] = mapped_column(String(32))
|
||
epochs: Mapped[int] = mapped_column(Integer)
|
||
patience: Mapped[int] = mapped_column(Integer)
|
||
best_pt: Mapped[str] = mapped_column(String(255), nullable=False)
|
||
last_pt: Mapped[str] = mapped_column(String(255), nullable=False)
|
||
|
||
|
||
class ProjectDetect(DbCommon):
|
||
"""
|
||
项目推理集合
|
||
"""
|
||
__tablename__ = "project_detect"
|
||
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))
|
||
detect_name: Mapped[str] = mapped_column(String(64), nullable=False)
|
||
train_id: Mapped[int] = mapped_column(Integer, nullable=False)
|
||
train_version: Mapped[str] = mapped_column(String(10))
|
||
pt_type: Mapped[str] = mapped_column(String(10))
|
||
pt_url: Mapped[str] = mapped_column(String(255))
|
||
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)
|