Boen_Shi
6a2e046884
- 新增 analyze、analyze_result、analyze_status 和 health 路由 - 实现图像上传和任务提交功能 - 添加任务状态查询和结果获取接口 - 集成 segformer 和 yolo 模型进行图像检测 - 实现 SAM3 预处理功能用于图像预处理判断 - 添加模型选择配置支持 segformer 和 yolo - 实现任务队列管理和异步处理机制 - 添加 Dockerfile 用于容器化部署 - 配置环境变量和 gitignore 规则 - 创建数据模型定义 API 响应结构
30 lines
853 B
Docker
30 lines
853 B
Docker
FROM python:3.11
|
|
|
|
# 安装系统依赖并清理缓存
|
|
RUN apt-get update && apt-get install -y \
|
|
libgl1 \
|
|
libglib2.0-0 \
|
|
libsm6 \
|
|
libxrender1 \
|
|
libxext6 \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
# 设置工作目录
|
|
WORKDIR /code
|
|
|
|
# 复制并安装 Python 依赖
|
|
COPY requirements.txt /code/requirements.txt
|
|
|
|
# 安装 Python 依赖（加速源）
|
|
RUN pip install --no-cache-dir --upgrade -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple \
|
|
&& pip install --no-cache-dir --upgrade torch torchvision --index-url https://download.pytorch.org/whl/cu130
|
|
|
|
# 复制应用代码
|
|
COPY ./app /code/app
|
|
|
|
# 删除无用的文件，避免占用磁盘空间
|
|
RUN rm -rf /code/app/core/*.onnx /code/app/core/*.data /code/app/core/*.pt
|
|
|
|
# 暴露端口并启动应用
|
|
CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "80"]
|