diff --git a/download_resource.bat b/download_resource.bat index 55059a6..dd2d37f 100644 --- a/download_resource.bat +++ b/download_resource.bat @@ -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 所有模型文件下载完成! +