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

Update download_resource.bat

Browse Source
This commit is contained in:
Jeffreytsai1004 2025-02-15 20:51:25 +08:00
parent 8da2d4273f
commit 17681d74f9
1 changed files with 60 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

86
download_resource.bat
View File

@ -70,46 +70,80 @@ echo ************************************************************
REM 下载 Neural Haircut 文件
cd NeuralHaircut
echo 正在下载Neural Haircut文件...
python -m gdown --folder "https://drive.google.com/drive/folders/1TCdJ0CKR3Q6LviovndOkJaKm8S1T9F_8"
cd diffusion_prior
python -m gdown "1_9EOUXHayKiGH5nkrayncln3d6m1uV7f"
cd ..\PIXIE
python -m gdown "1mPcGu62YPc4MdkT8FFiOCP629xsENHZf"
REM 使用7-Zip解压tar.gz文件
IF NOT EXIST pixie_data.tar.gz (
echo 错误：下载pixie_data.tar.gz失败
exit /b 1
IF EXIST "diffusion_prior\model.pt" (
echo Neural Haircut模型已存在，跳过下载...
) ELSE (
python -m gdown --folder "https://drive.google.com/drive/folders/1TCdJ0CKR3Q6LviovndOkJaKm8S1T9F_8"
)
cd diffusion_prior
IF EXIST "model.pt" (
echo 扩散先验模型已存在，跳过下载...
) ELSE (
python -m gdown "1_9EOUXHayKiGH5nkrayncln3d6m1uV7f"
)
cd ..\PIXIE
IF EXIST "pixie_data.tar" (
echo PIXIE模型已存在，跳过下载...
) ELSE (
python -m gdown "1mPcGu62YPc4MdkT8FFiOCP629xsENHZf"
REM 使用7-Zip解压tar.gz文件
IF NOT EXIST pixie_data.tar.gz (
echo 错误：下载pixie_data.tar.gz失败
exit /b 1
)
7z x pixie_data.tar.gz -y
7z x pixie_data.tar -y
del pixie_data.tar.gz
del pixie_data.tar
)
7z x pixie_data.tar.gz -y
7z x pixie_data.tar -y
del pixie_data.tar.gz
del pixie_data.tar
REM 下载 Matte-Anything 文件
cd ..\..\Matte-Anything
echo 正在下载Matte-Anything文件...
curl -L --retry 5 -o sam_vit_h_4b8939.pth https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth
curl -L --retry 5 -o groundingdino_swint_ogc.pth https://github.com/IDEA-Research/GroundingDINO/releases/download/v0.1.0-alpha/groundingdino_swint_ogc.pth
python -m gdown "1d97oKuITCeWgai2Tf3iNilt6rMSSYzkW" -O model.pth --retry-times 5
IF EXIST "sam_vit_h_4b8939.pth" (
echo SAM模型已存在，跳过下载...
) ELSE (
curl -L --retry 5 -o sam_vit_h_4b8939.pth https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth
)
IF EXIST "groundingdino_swint_ogc.pth" (
echo GroundingDINO模型已存在，跳过下载...
) ELSE (
curl -L --retry 5 -o groundingdino_swint_ogc.pth https://github.com/IDEA-Research/GroundingDINO/releases/download/v0.1.0-alpha/groundingdino_swint_ogc.pth
)
IF EXIST "model.pth" (
echo Matte-Anything模型已存在，跳过下载...
) ELSE (
python -m gdown "1d97oKuITCeWgai2Tf3iNilt6rMSSYzkW" -O model.pth
)
REM 下载 OpenPose 模型
cd ..\openpose
echo 正在下载OpenPose模型...
python -m gdown "1Yn03cKKfVOq4qXmgBMQD20UMRRRkd_tV" -O models.tar.gz
IF NOT EXIST models.tar.gz (
echo ERROR: Failed to download models.tar.gz
exit /b 1
IF EXIST "models\pose_iter_584000.caffemodel" (
echo OpenPose模型已存在，跳过下载...
) ELSE (
python -m gdown "1Yn03cKKfVOq4qXmgBMQD20UMRRRkd_tV" -O models.tar.gz
IF NOT EXIST models.tar.gz (
echo 错误：下载models.tar.gz失败
exit /b 1
)
7z x models.tar.gz -y
7z x models.tar -y -omodels
del models.tar.gz
del models.tar
)
7z x models.tar.gz -y
7z x models.tar -y -omodels
del models.tar.gz
del models.tar
REM 下载 hyperIQA 模型
cd ..\hyperIQA\pretrained
echo 正在下载hyperIQA模型...
python -m gdown "1OOUmnbvpGea0LIGpIWEbOyxfWx6UCiiE"
IF EXIST "hyperIQA.pth" (
echo hyperIQA模型已存在，跳过下载...
) ELSE (
python -m gdown "1OOUmnbvpGea0LIGpIWEbOyxfWx6UCiiE"
)
cd ..\..
echo.