From 6eb8253a00afb5cb4b408c3b7f680dfdb82b27f7 Mon Sep 17 00:00:00 2001
From: Jeffreytsai1004 <jeffreytsai1004@gmail.com>
Date: Sun, 23 Feb 2025 22:11:30 +0800
Subject: [PATCH] Update install.bat

---
 install.bat | 30 +++++++++++++++---------------
 1 file changed, 15 insertions(+), 15 deletions(-)

diff --git a/install.bat b/install.bat
index d54576f..d60f7b7 100644
--- a/install.bat
+++ b/install.bat
@@ -3,12 +3,12 @@ setlocal enabledelayedexpansion
 
 REM Set environment variables
 @CALL set "PYTHONDONTWRITEBYTECODE=1"
-@CALL set "GDOWN_CACHE=%PROJECT_DIR%\cache\gdown"
-@CALL set "TORCH_HOME=%PROJECT_DIR%\cache\torch"
-@CALL set "HF_HOME=%PROJECT_DIR%\cache\huggingface"
+@CALL set "GDOWN_CACHE=%~dp0\cache\gdown"
+@CALL set "TORCH_HOME=%~dp0\cache\torch"
+@CALL set "HF_HOME=%~dp0\cache\huggingface"
 
 @CALL "%~dp0micromamba.exe" shell init --shell cmd.exe --prefix "%~dp0\"
-@CALL SET PROJECT_DIR="%~dp0"
+@CALL SET PROJECT_DIR=%~dp0
 @CALL SET CUDA_HOME="C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8"
 @CALL SET BLENDER_DIR="C:\Program Files\Blender Foundation\Blender 3.6"
 @CALL SET VS_DIR="C:\Program Files (x86)\Microsoft Visual Studio\2019\BuildTools"
@@ -89,7 +89,7 @@ REM Create matte_anything environment
 @CALL python -m gdown 1d97oKuITCeWgai2Tf3iNilt6rMSSYzkW
 
 REM Create openpose environment
-@CALL cd %PROJECT_DIR%\ext\openpose
+@CALL cd %PROJECT_DIR%ext\openpose
 @CALL python -m gdown 1Yn03cKKfVOq4qXmgBMQD20UMRRRkd_tV
 @CALL tar -xvzf models.tar.gz
 @CALL git submodule update --init --recursive --remote
@@ -98,27 +98,27 @@ REM Create openpose environment
 @CALL "%~dp0condabin\micromamba.bat" activate openpose
 
 REM 完全清理构建目录
-@CALL cd %PROJECT_DIR%\ext\openpose
+@CALL cd %PROJECT_DIR%ext\openpose
 @CALL rmdir /s /q build 2>nul
 @CALL del /f /q CMakeCache.txt 2>nul
 @CALL rmdir /s /q CMakeFiles 2>nul
 @CALL mkdir build
-@CALL cd build
-
-@CALL set VS_VCVARS_FIXED="%VS_VCVARS:"=%"
-@CALL %VS_VCVARS_FIXED%
 
 REM 设置环境变量
-@CALL set "OPENPOSE_ROOT=%PROJECT_DIR%\ext\openpose"
-@CALL set "PYTHON_ROOT=%~dp0micromamba\envs\openpose"
+@CALL set "OPENPOSE_ROOT=%PROJECT_DIR%ext\openpose"
+@CALL set "PYTHON_ROOT=%PROJECT_DIR%micromamba\envs\openpose"
 @CALL set "PYTHON_EXECUTABLE=%PYTHON_ROOT%\python.exe"
 @CALL set "PYTHON_LIBRARY=%PYTHON_ROOT%\libs\python39.lib"
 @CALL set "PYTHON_INCLUDE_DIR=%PYTHON_ROOT%\include"
 @CALL set "OpenCV_DIR=%PYTHON_ROOT%\Library\cmake\opencv4"
 
-REM 确保在正确的目录执行 CMake
-@CALL cd %PROJECT_DIR%\ext\openpose\build
-@CALL cmake %PROJECT_DIR%\ext\openpose -G "Visual Studio 16 2019" -A x64 -T host=x64 ^
+REM 设置 VS 环境
+@CALL set VS_VCVARS_FIXED="%VS_VCVARS:"=%"
+@CALL %VS_VCVARS_FIXED%
+
+REM 执行 CMake 构建
+@CALL cd %OPENPOSE_ROOT%\build
+@CALL cmake %OPENPOSE_ROOT% -G "Visual Studio 16 2019" -A x64 -T host=x64 ^
     -DBUILD_PYTHON=true ^
     -DUSE_CUDNN=off ^
     -DOpenCV_DIR="%OpenCV_DIR%" ^