Boen_Shi
fd590d1294
- 在MaskInfo模型中添加score字段用于存储检测置信度 - 修改YOLO检测逻辑以提取和传递预测分数 - 更新坐标数据结构以包含置信度信息 - 调整数据处理流程以正确传输分数数据 - 修改Dockerfile以支持代码持久化部署 - 更新README文档说明代码持久化配置方式
34 lines
580 B
Python
34 lines
580 B
Python
from typing import List, Optional
|
|
from pydantic import BaseModel
|
|
|
|
""" 获取任务结果响应结构 """
|
|
|
|
|
|
class ImageInfo(BaseModel):
|
|
origin: str
|
|
image: str
|
|
|
|
|
|
class MaskInfo(BaseModel):
|
|
name: str
|
|
score: float
|
|
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
|