ilaandClaude Sonnet 4.6
68e4f444c5
- engine/defect_classes: HRIPCB 6类缺陷定义 + 4类扩展 + severity排序 - engine/base: DefectBox / DetectResult 数据类 + BaseInferenceEngine 抽象基类 - engine/ultralytics_adapter: YOLOv8 .pt 推理适配器 + warmup - engine/annotator: 按 severity 着色的标注图生成 + base64 编码 - engine/loader: 单例管理 + asyncio.Lock 推理串行化 + 热重载支持 - config: Pydantic v1 BaseSettings，读取 .env - schema: DetectResponse / HealthResponse / ReloadRequest 等响应模型 - api/detect: POST /api/detect，内存读图，推理锁保护 - api/health: GET /api/health - api/model: POST /api/model/reload 热重载接口 - main: FastAPI lifespan 启动加载模型 测试通过：/api/health 200，/api/detect 空图返回 pass，标注图 base64 正常 Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
27 lines
642 B
Python
27 lines
642 B
Python
from contextlib import asynccontextmanager
|
|
|
|
from fastapi import FastAPI
|
|
|
|
from config import settings
|
|
from engine.loader import init_engine
|
|
from api.detect import router as detect_router
|
|
from api.health import router as health_router
|
|
from api.model import router as model_router
|
|
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(app: FastAPI):
|
|
init_engine(settings.model_path, settings.runtime)
|
|
yield
|
|
|
|
|
|
app = FastAPI(
|
|
title="明析推理服务",
|
|
version="1.0.0",
|
|
lifespan=lifespan,
|
|
)
|
|
|
|
app.include_router(detect_router, prefix="/api")
|
|
app.include_router(health_router, prefix="/api")
|
|
app.include_router(model_router, prefix="/api")
|