From f4f28f9ba1e6c4e7b3a3d89e955bb99a93fc4f97 Mon Sep 17 00:00:00 2001 From: Boen_Shi Date: Thu, 29 Jan 2026 13:25:43 +0800 Subject: [PATCH] =?UTF-8?q?refactor(model):=20=E6=9B=B4=E6=96=B0=E6=A8=A1?= =?UTF-8?q?=E5=9E=8B=E6=96=87=E4=BB=B6=EF=BC=8C=E4=BC=98=E5=8C=96Dockerfil?= =?UTF-8?q?e=E5=92=8CREADME=E6=96=87=E6=A1=A3=E7=BB=93=E6=9E=84?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 更新yolo_detect的模型权重文件 - 移除Python依赖安装中的清华源加速配置 - 修改Dockerfile中删除文件的注释描述 - 重构README文档,简化使用教程内容 - 更新Docker构建和启动命令格式 - 添加GIT LFS相关提示信息 - 移除重复的持久化配置说明 --- Dockerfile | 4 +- README.md | 73 ++++++------------- .../model_data/best_epoch_weights.pth | 2 +- 3 files changed, 26 insertions(+), 53 deletions(-) diff --git a/Dockerfile b/Dockerfile index eeb61bd..e339ed4 100644 --- a/Dockerfile +++ b/Dockerfile @@ -15,14 +15,14 @@ WORKDIR /code # 复制并安装 Python 依赖 COPY requirements.txt /code/requirements.txt -# 安装 Python 依赖(加速源) +# 安装 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 # 暴露端口并启动应用 diff --git a/README.md b/README.md index bc6d100..75aeea4 100644 --- a/README.md +++ b/README.md @@ -1,56 +1,29 @@ ### Wall Docker 镜像使用教程 ---- +#### 构建方式 + +```bash +docker build -t wall . +``` + +#### 启动方式 > 采用模型文件持久化,方便后续更新模型而不需要重新创建容器,以及统一配置配置文件 +> +> 后续更新,如果只更新了核心文件,则仅需git pull后重新启动容器 +> +> 如果更新了代码文件,则需要重新构建镜像 -1. 导入docker images `docker load -i wall.tar` +```bash +sudo docker run -d \ + --name [docker_container_name] \ + --gpus all \ + -p [local_port]:80 \ + -v $(pwd)/uploads:/code/uploads \ + -v $(pwd)/app/core:/code/app/core \ + -v $(pwd)/.env:/code/.env \ + wall +``` -2. 进入持久化目录,新建.env,添加相应内容 - - ```env - UPLOAD_DIR=uploads - MOCK=false - MODEL=segformer #segformer, yolo,目前打包模型只有segformer - ``` - -3. 解压算法模型目录到core文件夹 - - ```bash - tar -xvf core.tar - ``` - -4. 使用指令运行docker镜像 - - ```bash - sudo docker run -d \ - --name [docker_container_name] \ - --gpus all \ - -p [local_port]:80 \ - -v $(pwd)/uploads:/code/uploads \ - -v $(pwd)/core:/code/app/core \ - -v $(pwd)/.env:/code/.env \ - wall - ``` - -5. 如果后续需要更新模型,只需要覆盖掉core内的文件,更改.env配置文件后,即可继续运行 - ---- - -> 如果不想要模型文件持久化,则不需要解压算法文件了 - -1. 导入docker images `docker load -i wall.tar` - -2. 使用指令运行docker镜像 - - ```bash - sudo docker run -d \ - --name [docker_container_name] \ - --gpus all \ - -p [local_port]:80 \ - -v $(pwd)/core:/code/app/core \ - -v $(pwd)/.env:/code/.env \ - wall - ``` - - \ No newline at end of file +> TIPS:由于部分文件采用GIT LFS 管理,请先安装GIT LFS +> 在clone或者pull时,建议先clone代码文件,然后停掉,再用git lfs pull可查看大文件下载进度 \ No newline at end of file diff --git a/app/core/yolo_detect/model_data/best_epoch_weights.pth b/app/core/yolo_detect/model_data/best_epoch_weights.pth index 5ad9cf9..90711c9 100644 --- a/app/core/yolo_detect/model_data/best_epoch_weights.pth +++ b/app/core/yolo_detect/model_data/best_epoch_weights.pth @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:1fab1422f9b5500cd5941a0f6b4445d366937637b3d8615f931f2092aa1f76d5 +oid sha256:04f94d34d6965ea6df5599b3825bf256f863727b31288fb744b94f1f99a64c29 size 174941676