50 lines
1.8 KiB
Batchfile
50 lines
1.8 KiB
Batchfile
@echo off
|
|
setlocal enabledelayedexpansion
|
|
|
|
REM 设置环境变量
|
|
set "PROJECT_DIR=%~dp0"
|
|
set "PYTHONDONTWRITEBYTECODE=1"
|
|
set "GDOWN_CACHE=cache\gdown"
|
|
set "TORCH_HOME=cache\torch"
|
|
set "HF_HOME=cache\huggingface"
|
|
|
|
REM 创建缓存目录
|
|
mkdir "%PROJECT_DIR%\cache\gdown" 2>nul
|
|
mkdir "%PROJECT_DIR%\cache\torch" 2>nul
|
|
mkdir "%PROJECT_DIR%\cache\huggingface" 2>nul
|
|
|
|
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
|
|
|
|
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
|
|
|
|
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
|
|
|
|
cd "%PROJECT_DIR%\resource\openpose"
|
|
python -m gdown 1Yn03cKKfVOq4qXmgBMQD20UMRRRkd_tV
|
|
tar -xf models.tar.gz
|
|
|
|
cd "%PROJECT_DIR%"
|