diff --git a/download_resource.bat b/download_resource.bat index 0e868c7..fb2fe98 100644 --- a/download_resource.bat +++ b/download_resource.bat @@ -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 模型检查 diff --git a/install.bat b/install.bat index 470247d..704fddd 100644 --- a/install.bat +++ b/install.bat @@ -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! diff --git a/run.bat b/run.bat index 94284c2..81aa267 100644 --- a/run.bat +++ b/run.bat @@ -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%" (