Boen_Shi
b054769b40
- 新增yolo_detect模块，包含backbone、nets、utils等组件 - 在模型配置中添加yolo_detect选项，支持新的检测模型 - 移除SAM3预处理相关代码和配置项 - 更新Dockerfile删除core目录下所有文件以减少镜像体积 - 修改worker服务移除图像标签预处理逻辑，直接进行模型检测
30 lines
804 B
Docker
30 lines
804 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
|
|
|
|
# 暴露端口并启动应用
|
|
CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "80"]
|