install.bat

@@ -96,20 +96,47 @@ REM Create openpose environment
 @CALL cd %PROJECT_DIR%
 @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 %PROJECT_DIR%ext\openpose\build
-@CALL cmake %PROJECT_DIR%ext\openpose -G "Visual Studio 16 2019" -A x64 -T host=x64
+
+REM 设置环境变量
+@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 设置 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%" ^
+    -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
 
 REM Create pixie-env environment
 @CALL cd %PROJECT_DIR%
-@CALL "%~dp0micromamba.exe" create -n pixie-env python=3.9 ^
+@CALL "%~dp0micromamba.exe" create -n pixie-env python=3.8 ^
     -c pytorch -c nvidia -c fvcore -c conda-forge -c pytorch3d -r "%~dp0\" -y
 @CALL "%~dp0condabin\micromamba.bat" activate pixie-env
 @CALL cd %PROJECT_DIR%\ext\PIXIE\