Update
This commit is contained in:
parent
6ebe457f94
commit
e40a80ce47
@ -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%"
|
||||
|
||||
15
install.bat
15
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
|
||||
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user