Boen_Shi
6a2e046884
- 新增 analyze、analyze_result、analyze_status 和 health 路由 - 实现图像上传和任务提交功能 - 添加任务状态查询和结果获取接口 - 集成 segformer 和 yolo 模型进行图像检测 - 实现 SAM3 预处理功能用于图像预处理判断 - 添加模型选择配置支持 segformer 和 yolo - 实现任务队列管理和异步处理机制 - 添加 Dockerfile 用于容器化部署 - 配置环境变量和 gitignore 规则 - 创建数据模型定义 API 响应结构
33 lines
563 B
Python
33 lines
563 B
Python
from typing import List, Optional
|
|
from pydantic import BaseModel
|
|
|
|
""" 获取任务结果响应结构 """
|
|
|
|
|
|
class ImageInfo(BaseModel):
|
|
origin: str
|
|
image: str
|
|
|
|
|
|
class MaskInfo(BaseModel):
|
|
name: str
|
|
coords: List[List[int]]
|
|
|
|
|
|
class ResultItem(BaseModel):
|
|
id: str
|
|
images: ImageInfo
|
|
masks: List[MaskInfo]
|
|
|
|
|
|
class AnalyzeResultData(BaseModel):
|
|
taskId: str
|
|
status: str
|
|
completedAt: Optional[str] = None
|
|
results: Optional[List[ResultItem]] = None
|
|
|
|
|
|
class AnalyzeResult(BaseModel):
|
|
success: bool
|
|
data: AnalyzeResultData
|