Update download_resource.bat

This commit is contained in:
Jeffreytsai1004 2025-02-16 18:38:41 +08:00
parent 41f8e0efab
commit 19b5d71982

View File

@ -6,6 +6,10 @@ IF EXIST "resource" (
echo 清理旧的资源目录...
rmdir /s /q "resource"
)
else (
echo 资源目录不存在,创建资源目录...
mkdir resource
)
@ -83,21 +87,26 @@ mkdir %PROJECT_DIR%/resource/openpose
REM Download Neural Haircut files
echo 正在下载Neural Haircut文件...
mkdir %PROJECT_DIR%/resource/NeuralHaircut/
cd %PROJECT_DIR%/resource/NeuralHaircut
python -m gdown --folder https://drive.google.com/drive/folders/1TCdJ0CKR3Q6LviovndOkJaKm8S1T9F_8
REM 更新 diffusion_prior 模型
cd %PROJECT_DIR%/resource/NeuralHaircut/pretrained_models/diffusion_prior
echo 正在下载Neural Haircut扩散先验模型文件...
python -m gdown 1_9EOUXHayKiGH5nkrayncln3d6m1uV7f
REM 下载 PIXIE 文件
cd %PROJECT_DIR%/resource/NeuralHaircut/PIXIE
echo 正在下载Neural HaircutPIXIE模型文件...
python -m gdown 1mPcGu62YPc4MdkT8FFiOCP629xsENHZf && tar -xvzf pixie_data.tar.gz ./ && rm pixie_data.tar.gz
REM 下载 hyperIQA 模型
cd %PROJECT_DIR%/resource/hyperIQA/pretrained
echo 正在下载hyperIQA模型文件...
python -m gdown 1OOUmnbvpGea0LIGpIWEbOyxfWx6UCiiE
cd %PROJECT_DIR%
REM Download Matte-Anything files
mkdir %PROJECT_DIR%/resource/Matte-Anything/pretrained
cd %PROJECT_DIR%/resource/Matte-Anything/pretrained
echo 正在下载Matte-Anything扩散先验模型文件...
wget https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth
@ -106,11 +115,14 @@ wget https://github.com/IDEA-Research/GroundingDINO/releases/download/v0.1.0-alp
echo 正在下载Matte-Anything模型文件...
python -m gdown 1d97oKuITCeWgai2Tf3iNilt6rMSSYzkW
# Download OpenPose files
REM Download OpenPose files
cd %PROJECT_DIR%/resource/openpose
echo 正在下载OpenPose模型文件...
python -m gdown 1Yn03cKKfVOq4qXmgBMQD20UMRRRkd_tV && tar -xvzf models.tar.gz && rm models.tar.gz
cd %PROJECT_DIR%
echo 所有模型文件下载完成!