CGNICO/GaussianHaircut
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

更新 download_resource.bat

Browse Source
This commit is contained in:
Jeffrey 2025-02-18 22:12:39 +08:00
parent 810ff3ffb1
commit b8c941df04
1 changed files with 26 additions and 48 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

74
download_resource.bat
View File

@ -1,55 +1,33 @@
@echo off
chcp 437 >nul
setlocal enabledelayedexpansion
REM Set environment variables
set "PROJECT_DIR=%~dp0"
set "PYTHONDONTWRITEBYTECODE=1"
set "GDOWN_CACHE=%PROJECT_DIR%\cache\gdown"
set "TORCH_HOME=%PROJECT_DIR%\cache\torch"
set "HF_HOME=%PROJECT_DIR%\cache\huggingface"
@CALL SET PROJECT_DIR=%~dp0
REM Check Python
python --version >nul 2>&1
if %ERRORLEVEL% neq 0 (
echo ERROR: Python not found
exit /b 1
)
@CALL python -m pip install --upgrade pip
@CALL python -m pip install gdown
REM Create directories
mkdir "%PROJECT_DIR%\cache\gdown" 2>nul
mkdir "%PROJECT_DIR%\cache\torch" 2>nul
mkdir "%PROJECT_DIR%\cache\huggingface" 2>nul
mkdir "%PROJECT_DIR%\resource\NeuralHaircut\pretrained_models\diffusion_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
@CALL mkdir %PROJECT_DIR%\resource\NeuralHaircut\PIXIE
@CALL mkdir %PROJECT_DIR%\resource\hyperIQA\pretrained
@CALL mkdir %PROJECT_DIR%\resource\Matte-Anything\pretrained
@CALL mkdir %PROJECT_DIR%\resource\openpose
REM Install gdown
python -m pip install --upgrade pip gdown
@CALL cd %PROJECT_DIR%\resource\NeuralHaircut
@CALL python -m gdown --folder https://drive.google.com/drive/folders/1TCdJ0CKR3Q6LviovndOkJaKm8S1T9F_8
@CALL cd %PROJECT_DIR%\resource\NeuralHaircut\pretrained_models\diffusion_prior
@CALL python -m gdown 1_9EOUXHayKiGH5nkrayncln3d6m1uV7f
@CALL cd %PROJECT_DIR%\resource\NeuralHaircut\PIXIE
@CALL python -m gdown 1mPcGu62YPc4MdkT8FFiOCP629xsENHZf
@CALL tar -xvzf pixie_data.tar.gz ./
@CALL rm pixie_data.tar.gz
@CALL cd %PROJECT_DIR%\resource\hyperIQA\pretrained
@CALL python -m gdown 1OOUmnbvpGea0LIGpIWEbOyxfWx6UCiiE
@CALL cd %PROJECT_DIR%
REM Download model files
cd "%PROJECT_DIR%\resource\NeuralHaircut"
python -m gdown --folder https://drive.google.com/drive/folders/1TCdJ0CKR3Q6LviovndOkJaKm8S1T9F_8 || exit /b 1
@CALL cd %PROJECT_DIR%\resource\Matte-Anything\pretrained
@CALL curl -L -o sam_vit_h_4b8939.pth https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth
@CALL curl -L -o groundingdino_swint_ogc.pth https://github.com/IDEA-Research/GroundingDINO/releases/download/v0.1.0-alpha/groundingdino_swint_ogc.pth
@CALL python -m gdown 1d97oKuITCeWgai2Tf3iNilt6rMSSYzkW
cd "%PROJECT_DIR%\resource\NeuralHaircut\pretrained_models\diffusion_prior"
python -m gdown 1_9EOUXHayKiGH5nkrayncln3d6m1uV7f || exit /b 1
cd "%PROJECT_DIR%\resource\NeuralHaircut\PIXIE"
python -m gdown 1mPcGu62YPc4MdkT8FFiOCP629xsENHZf || exit /b 1
cd "%PROJECT_DIR%\resource\hyperIQA\pretrained"
python -m gdown 1OOUmnbvpGea0LIGpIWEbOyxfWx6UCiiE || exit /b 1
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 || exit /b 1
curl -L https://github.com/IDEA-Research/GroundingDINO/releases/download/v0.1.0-alpha/groundingdino_swint_ogc.pth -o groundingdino_swint_ogc.pth || exit /b 1
python -m gdown 1d97oKuITCeWgai2Tf3iNilt6rMSSYzkW || exit /b 1
cd "%PROJECT_DIR%\resource\openpose"
python -m gdown 1Yn03cKKfVOq4qXmgBMQD20UMRRRkd_tV || exit /b 1
tar -xf models.tar.gz || exit /b 1
cd "%PROJECT_DIR%"
echo All resources downloaded successfully!
@CALL cd %PROJECT_DIR%\resource\openpose
@CALL python -m gdown 1Yn03cKKfVOq4qXmgBMQD20UMRRRkd_tV
@CALL tar -xvzf models.tar.gz
@CALL rm models.tar.gz