Boen_Shi
6a2e046884
- 新增 analyze、analyze_result、analyze_status 和 health 路由 - 实现图像上传和任务提交功能 - 添加任务状态查询和结果获取接口 - 集成 segformer 和 yolo 模型进行图像检测 - 实现 SAM3 预处理功能用于图像预处理判断 - 添加模型选择配置支持 segformer 和 yolo - 实现任务队列管理和异步处理机制 - 添加 Dockerfile 用于容器化部署 - 配置环境变量和 gitignore 规则 - 创建数据模型定义 API 响应结构
17 lines
491 B
Python
17 lines
491 B
Python
from app.core.segformer.detect import Detection as SegFormer, DetectionMock as SegFormerMock
|
|
from app.core.yolo.detect import YOLOSeg
|
|
|
|
|
|
class Model:
|
|
def __init__(self):
|
|
from app.main import MODEL
|
|
if MODEL == "segformer":
|
|
print("使用 SegFormer 模型")
|
|
self.detection = SegFormer()
|
|
elif MODEL == "yolo":
|
|
print("使用 YOLO 模型")
|
|
self.detection = YOLOSeg()
|
|
|
|
def getModel(self):
|
|
return self.detection
|