2025-02-17 00:55:28 +08:00
|
|
|
|
@echo off
|
2025-02-17 23:40:30 +08:00
|
|
|
|
|
|
|
|
|
|
REM 设置环境变量
|
|
|
|
|
|
SET CUDA_HOME="C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8"
|
|
|
|
|
|
SET PATH="%CUDA_HOME%\bin;%PATH%"
|
|
|
|
|
|
SET BLENDER_DIR="C:\Program Files\Blender Foundation\Blender 3.6"
|
|
|
|
|
|
SET VS_DIR="C:\Program Files (x86)\Microsoft Visual Studio\2019\BuildTools"
|
|
|
|
|
|
SET VS_VCVARS="%VS_DIR%\VC\Auxiliary\Build\vcvars64.bat"
|
|
|
|
|
|
SET PROJECT_DIR=%~dp0
|
|
|
|
|
|
SET MICROMAMBA_EXE=%~dp0micromamba.exe
|
|
|
|
|
|
SET MAMBA_ROOT_PREFIX=%PROJECT_DIR%\micromamba
|
|
|
|
|
|
SET PYTHONDONTWRITEBYTECODE=1
|
|
|
|
|
|
SET GDOWN_CACHE=cache\gdown
|
|
|
|
|
|
SET TORCH_HOME=cache\torch
|
|
|
|
|
|
SET HF_HOME=cache\huggingface
|
2025-02-17 00:55:28 +08:00
|
|
|
|
|
2025-02-17 08:44:22 +08:00
|
|
|
|
REM 检查是否安装了python
|
|
|
|
|
|
if not exist "%PROJECT_DIR%\venv\Scripts\python.exe" (
|
|
|
|
|
|
echo 错误：未找到python.exe
|
|
|
|
|
|
echo 请确保python已安装并配置好环境变量
|
|
|
|
|
|
exit /b 1
|
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
REM pip安装检查更新
|
|
|
|
|
|
python -m pip install --upgrade pip
|
|
|
|
|
|
|
|
|
|
|
|
REM 安装gdown
|
|
|
|
|
|
python -m pip install gdown
|
|
|
|
|
|
|
2025-02-17 00:55:28 +08:00
|
|
|
|
mkdir %PROJECT_DIR%\resource\NeuralHaircut\
|
|
|
|
|
|
cd %PROJECT_DIR%\resource\NeuralHaircut\
|
2025-02-17 00:58:08 +08:00
|
|
|
|
python -m gdown --folder https://drive.google.com/drive/folders/1TCdJ0CKR3Q6LviovndOkJaKm8S1T9F_8
|
2025-02-17 00:55:28 +08:00
|
|
|
|
cd %PROJECT_DIR%\resource\NeuralHaircut\pretrained_models\diffusion_prior
|
|
|
|
|
|
python -m gdown 1_9EOUXHayKiGH5nkrayncln3d6m1uV7f
|
|
|
|
|
|
mkdir %PROJECT_DIR%\resource\NeuralHaircut\PIXIE
|
|
|
|
|
|
cd %PROJECT_DIR%\resource\NeuralHaircut\PIXIE
|
2025-02-17 08:31:59 +08:00
|
|
|
|
python -m gdown 1mPcGu62YPc4MdkT8FFiOCP629xsENHZf
|
2025-02-17 08:44:22 +08:00
|
|
|
|
7z x pixie_data.tar.gz -o./ && tar -xvzf pixie_data.tar.gz
|
2025-02-17 00:55:28 +08:00
|
|
|
|
mkdir %PROJECT_DIR%\resource\hyperIQA\pretrained
|
|
|
|
|
|
cd %PROJECT_DIR%\resource\hyperIQA\pretrained
|
|
|
|
|
|
python -m gdown 1OOUmnbvpGea0LIGpIWEbOyxfWx6UCiiE
|
|
|
|
|
|
cd %PROJECT_DIR%
|
|
|
|
|
|
|
|
|
|
|
|
mkdir %PROJECT_DIR%\resource\Matte-Anything\pretrained
|
|
|
|
|
|
cd %PROJECT_DIR%\resource\Matte-Anything\pretrained
|
|
|
|
|
|
wget https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth
|
|
|
|
|
|
wget https://github.com/IDEA-Research/GroundingDINO/releases/download/v0.1.0-alpha/groundingdino_swint_ogc.pth
|
|
|
|
|
|
python -m gdown 1d97oKuITCeWgai2Tf3iNilt6rMSSYzkW
|
|
|
|
|
|
|
|
|
|
|
|
mkdir %PROJECT_DIR%\resource\openpose\
|
|
|
|
|
|
cd %PROJECT_DIR%\resource\openpose\
|
2025-02-17 08:44:22 +08:00
|
|
|
|
python -m gdown 1Yn03cKKfVOq4qXmgBMQD20UMRRRkd_tV && tar -xvzf models.tar.gz
|
