Boen_Shi
fd590d1294
- 在MaskInfo模型中添加score字段用于存储检测置信度 - 修改YOLO检测逻辑以提取和传递预测分数 - 更新坐标数据结构以包含置信度信息 - 调整数据处理流程以正确传输分数数据 - 修改Dockerfile以支持代码持久化部署 - 更新README文档说明代码持久化配置方式
30 lines
844 B
Docker
30 lines
844 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"]
|