diff --git a/download_resource.bat b/download_resource.bat index 0b59929..aae5d36 100644 --- a/download_resource.bat +++ b/download_resource.bat @@ -1,52 +1,49 @@ @echo off +setlocal enabledelayedexpansion 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 +set "PROJECT_DIR=%~dp0" +set "PYTHONDONTWRITEBYTECODE=1" +set "GDOWN_CACHE=cache\gdown" +set "TORCH_HOME=cache\torch" +set "HF_HOME=cache\huggingface" -REM 检查是否安装了python -if not exist "%PROJECT_DIR%\venv\Scripts\python.exe" ( - echo 错误:未找到python.exe - echo 请确保python已安装并配置好环境变量 - exit /b 1 -) +REM 创建缓存目录 +mkdir "%PROJECT_DIR%\cache\gdown" 2>nul +mkdir "%PROJECT_DIR%\cache\torch" 2>nul +mkdir "%PROJECT_DIR%\cache\huggingface" 2>nul -REM pip安装检查更新 -python -m pip install --upgrade pip +REM 创建资源目录 +mkdir "%PROJECT_DIR%\resource\NeuralHaircut\pretrained_models\diffusion_prior" 2>nul +mkdir "%PROJECT_DIR%\resource\NeuralHaircut\pretrained_models\strand_prior" 2>nul +mkdir "%PROJECT_DIR%\resource\NeuralHaircut\PIXIE" 2>nul +mkdir "%PROJECT_DIR%\resource\Matte-Anything\pretrained" 2>nul +mkdir "%PROJECT_DIR%\resource\openpose\models" 2>nul +mkdir "%PROJECT_DIR%\resource\hyperIQA\pretrained" 2>nul REM 安装gdown python -m pip install gdown -mkdir %PROJECT_DIR%\resource\NeuralHaircut\ -cd %PROJECT_DIR%\resource\NeuralHaircut\ +REM 下载模型文件 +cd "%PROJECT_DIR%\resource\NeuralHaircut" python -m gdown --folder https://drive.google.com/drive/folders/1TCdJ0CKR3Q6LviovndOkJaKm8S1T9F_8 -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 -python -m gdown 1mPcGu62YPc4MdkT8FFiOCP629xsENHZf -7z x pixie_data.tar.gz -o./ && tar -xvzf pixie_data.tar.gz -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 +cd "%PROJECT_DIR%\resource\NeuralHaircut\pretrained_models\diffusion_prior" +python -m gdown 1_9EOUXHayKiGH5nkrayncln3d6m1uV7f + +cd "%PROJECT_DIR%\resource\NeuralHaircut\PIXIE" +python -m gdown 1mPcGu62YPc4MdkT8FFiOCP629xsENHZf + +cd "%PROJECT_DIR%\resource\hyperIQA\pretrained" +python -m gdown 1OOUmnbvpGea0LIGpIWEbOyxfWx6UCiiE + +cd "%PROJECT_DIR%\resource\Matte-Anything\pretrained" +curl -L https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth -o sam_vit_h_4b8939.pth +curl -L https://github.com/IDEA-Research/GroundingDINO/releases/download/v0.1.0-alpha/groundingdino_swint_ogc.pth -o groundingdino_swint_ogc.pth python -m gdown 1d97oKuITCeWgai2Tf3iNilt6rMSSYzkW -mkdir %PROJECT_DIR%\resource\openpose\ -cd %PROJECT_DIR%\resource\openpose\ -python -m gdown 1Yn03cKKfVOq4qXmgBMQD20UMRRRkd_tV && tar -xvzf models.tar.gz +cd "%PROJECT_DIR%\resource\openpose" +python -m gdown 1Yn03cKKfVOq4qXmgBMQD20UMRRRkd_tV +tar -xf models.tar.gz + +cd "%PROJECT_DIR%" diff --git a/install.bat b/install.bat index 9485d9e..66f5874 100644 --- a/install.bat +++ b/install.bat @@ -31,8 +31,15 @@ cd %PROJECT_DIR%\ext\kaolin && git checkout v0.15.0 cd %PROJECT_DIR%\ext && git clone https://github.com/SSL92/hyperIQA REM 安装环境 -cd %PROJECT_DIR% && %MICROMAMBA_EXE% create -n gaussian_splatting_hair -f environment.yml -CALL %MICROMAMBA_EXE% activate -p %MAMBA_ROOT_PREFIX%\envs\gaussian_splatting_hair +%MICROMAMBA_EXE% create -y -p %MAMBA_ROOT_PREFIX%\envs\gaussian_splatting_hair ^ + python=3.9 pip=23.3.1 setuptools=69.5.1 ^ + pytorch=2.1.1 torchvision=0.16.1 torchaudio=2.1.1 pytorch-cuda=11.8 ^ + cmake=3.28.0 pyhocon=0.3.60 icecream=2.1.3 einops=0.6.0 ^ + accelerate=0.18.0 jsonmerge=1.9.0 easydict=1.9 iopath=0.1.10 ^ + tensorboardx=2.6 scikit-image=0.20.0 fvcore=0.1.5 toml=0.10.2 tqdm=4.66.5 ^ + gdown=5.2.0 colmap=3.10 ^ + -c pytorch -c conda-forge -c defaults -c anaconda -c fvcore -c iopath -c bottler -c nvidia + REM 从resource文件夹拷贝Neural Haircut文件 xcopy /E /I /Y %PROJECT_DIR%\resource\NeuralHaircut %PROJECT_DIR%\ext\NeuralHaircut xcopy /E /I /Y %PROJECT_DIR%\resource\hyperIQA\pretrained %PROJECT_DIR%\ext\hyperIQA\pretrained @@ -114,4 +121,8 @@ pip install "git+https://github.com/1adrianb/face-alignment.git@54623537fd9618ca REM 退出pixie环境 CALL %MICROMAMBA_EXE% deactivate +REM 安装pip包 +pip install pysdf==0.1.9 clean-fid==0.1.35 face-alignment==1.4.1 clip==0.2.0 ^ + torchdiffeq==0.2.3 torchsde==0.2.5 resize-right==0.0.2 +