From b8c941df047bbbe7d7073f13a8d358b37e839234 Mon Sep 17 00:00:00 2001
From: Jeffrey <jeffreytsai1004@gmail.com>
Date: Tue, 18 Feb 2025 22:12:39 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20download=5Fresource.bat?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 download_resource.bat | 74 +++++++++++++++----------------------------
 1 file changed, 26 insertions(+), 48 deletions(-)

diff --git a/download_resource.bat b/download_resource.bat
index b2e8051..f91dec8 100644
--- a/download_resource.bat
+++ b/download_resource.bat
@@ -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
\ No newline at end of file