41 lines
1.8 KiB
Python
41 lines
1.8 KiB
Python
from sqlalchemy import Column, String, Text, Boolean, Float, Integer
|
||
from .base import BaseModel
|
||
import json
|
||
|
||
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格式")
|
||
|
||
def to_dict(self) -> dict:
|
||
"""将模型实例转换为字典格式"""
|
||
return {
|
||
"id": self.id,
|
||
"name": self.name,
|
||
"description": self.description,
|
||
"version": self.version,
|
||
"model_path": self.model_path,
|
||
"config_path": self.config_path,
|
||
"status": self.status,
|
||
"accuracy": self.accuracy,
|
||
"detection_classes": self.detection_classes,
|
||
"input_size": self.input_size,
|
||
"inference_time": self.inference_time,
|
||
"is_enabled": self.is_enabled,
|
||
"creator": self.creator,
|
||
"tags": self.tags,
|
||
"created_at": self.created_at,
|
||
"updated_at": self.updated_at
|
||
} |