This commit is contained in:
Jeffreytsai1004 2025-02-16 17:01:41 +08:00
parent 818304772b
commit 0a7148eb40
3 changed files with 35 additions and 10 deletions

View File

@ -143,22 +143,22 @@ IF EXIST "%PIXIE_PATH%" (
REM SAM 模型检查
IF EXIST "%SAM_PATH%" (
echo [√] %MODEL_SAM_NAME%已存在
echo [√] SAM模型已存在
SET /A EXISTING_FILES+=1
) ELSE (
echo [×] 缺少%MODEL_SAM_NAME%
echo [×] 缺少SAM模型
SET /A MISSING_FILES+=1
SET MISSING_LIST=!MISSING_LIST!%MODEL_SAM_NAME%,
SET MISSING_LIST=!MISSING_LIST!SAM模型,
)
REM GroundingDINO 模型检查
IF EXIST "%DINO_PATH%" (
echo [√] %MODEL_DINO_NAME%已存在
echo [√] GroundingDINO模型已存在
SET /A EXISTING_FILES+=1
) ELSE (
echo [×] 缺少%MODEL_DINO_NAME%
echo [×] 缺少GroundingDINO模型
SET /A MISSING_FILES+=1
SET MISSING_LIST=!MISSING_LIST!%MODEL_DINO_NAME%,
SET MISSING_LIST=!MISSING_LIST!GroundingDINO模型,
)
REM OpenPose 模型检查

View File

@ -91,6 +91,18 @@ IF %ERRORLEVEL% EQU 0 (
exit /b 1
)
REM 检查必要的环境变量
IF NOT DEFINED PROJECT_DIR (
echo 错误未设置PROJECT_DIR环境变量
exit /b 1
)
IF NOT DEFINED VS2019_VCVARS (
echo 错误未设置VS2019_VCVARS环境变量
echo 请设置VS2019_VCVARS指向Visual Studio 2019的vcvars64.bat
exit /b 1
)
REM 创建缓存目录
mkdir cache\gdown 2>nul
mkdir cache\torch 2>nul
@ -268,4 +280,17 @@ if %FREE_SPACE% LSS 21474836480 (
exit /b 1
)
REM 创建环境激活脚本
echo @echo off > activate_gaussian_splatting_hair.bat
echo CALL "%%MICROMAMBA_EXE%%" activate -p %%MAMBA_ROOT_PREFIX%%\envs\gaussian_splatting_hair >> activate_gaussian_splatting_hair.bat
echo @echo off > activate_matte_anything.bat
echo CALL "%%MICROMAMBA_EXE%%" activate -p %%MAMBA_ROOT_PREFIX%%\envs\matte_anything >> activate_matte_anything.bat
echo @echo off > activate_openpose.bat
echo CALL "%%MICROMAMBA_EXE%%" activate -p %%MAMBA_ROOT_PREFIX%%\envs\openpose >> activate_openpose.bat
echo @echo off > activate_pixie-env.bat
echo CALL "%%MICROMAMBA_EXE%%" activate -p %%MAMBA_ROOT_PREFIX%%\envs\pixie-env >> activate_pixie-env.bat
echo Installation completed!

View File

@ -16,12 +16,12 @@ REM 确保以下环境变量可用:
REM PROJECT_DIR 和 DATA_PATH
REM 检查必要的环境变量
IF "%PROJECT_DIR%"=="" (
echo ERROR: PROJECT_DIR environment variable is not set
IF NOT DEFINED PROJECT_DIR (
echo 错误未设置PROJECT_DIR环境变量
exit /b 1
)
IF "%DATA_PATH%"=="" (
echo ERROR: DATA_PATH environment variable is not set
IF NOT DEFINED DATA_PATH (
echo 错误未设置DATA_PATH环境变量
exit /b 1
)
IF NOT EXIST "%BLENDER_DIR%" (