19 lines
1.0 KiB
Python
19 lines
1.0 KiB
Python
|
from sqlalchemy import Column, String, Text, Boolean, Float, Integer
|
|||
|
|
from .base import BaseModel
|
|||
|
|
|
|||
|
|
class Algorithm(BaseModel):
|
|||
|
|
__tablename__ = "algorithms"
|
|||
|
|
|
|||
|
|
name = Column(String(100), nullable=False, comment="算法名称")
|
|||
|
|
description = Column(Text, comment="算法描述")
|
|||
|
|
version = Column(String(20), nullable=False, comment="算法版本")
|
|||
|
|
model_path = Column(String(255), comment="模型文件路径")
|
|||
|
|
config_path = Column(String(255), comment="配置文件路径")
|
|||
|
|
status = Column(String(20), default="inactive", comment="算法状态: active, inactive, training")
|
|||
|
|
accuracy = Column(Float, comment="算法准确率")
|
|||
|
|
detection_classes = Column(Text, comment="检测类别，JSON格式")
|
|||
|
|
input_size = Column(String(20), comment="输入尺寸，如: 640x640")
|
|||
|
|
inference_time = Column(Float, comment="推理时间(ms)")
|
|||
|
|
is_enabled = Column(Boolean, default=True, comment="是否启用")
|
|||
|
|
creator = Column(String(50), comment="创建者")
|
|||
|
|
tags = Column(Text, comment="标签，JSON格式")
|