Update install.bat

This commit is contained in:
Jeffreytsai1004 2025-02-23 19:27:21 +08:00
parent 849b60e602
commit 2e28625267

View File

@ -94,15 +94,42 @@ REM Create openpose environment
@CALL tar -xvzf models.tar.gz
@CALL git submodule update --init --recursive --remote
@CALL cd %PROJECT_DIR%
@CALL "%~dp0micromamba.exe" create -y -n openpose cmake=3.20 -c conda-forge -r "%~dp0\"
@CALL "%~dp0micromamba.exe" create -y -n openpose python=3.9 cmake=3.20 opencv=4.5.3 -c conda-forge -r "%~dp0\"
@CALL "%~dp0condabin\micromamba.bat" activate openpose
REM 完全清理构建目录
@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%
@CALL cmake .. -G "Visual Studio 16 2019" -A x64 -T host=x64 -DBUILD_PYTHON=true -DUSE_CUDNN=off
REM 设置环境变量
@CALL set "OPENPOSE_ROOT=%PROJECT_DIR%\ext\openpose"
@CALL set "PYTHON_ROOT=%~dp0micromamba\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 ^
-DBUILD_PYTHON=true ^
-DUSE_CUDNN=off ^
-DOpenCV_DIR="%OpenCV_DIR%" ^
-DPYTHON_EXECUTABLE="%PYTHON_EXECUTABLE%" ^
-DPYTHON_LIBRARY="%PYTHON_LIBRARY%" ^
-DPYTHON_INCLUDE_DIR="%PYTHON_INCLUDE_DIR%" ^
-DPYTHON_VERSION_MAJOR=3 ^
-DPYTHON_VERSION_MINOR=9 ^
-DPYTHON_DEBUG_LIBRARY="" ^
-DPYTHON_ROOT_DIR="%PYTHON_ROOT%"
@CALL cmake --build . --config Release
@CALL cd %PROJECT_DIR%
@CALL "%~dp0condabin\micromamba.bat" deactivate