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格式") |