Update

README.md

@@ -96,13 +96,18 @@
 
 5. **安装 7-Zip**
     - 从 https://7-zip.org/ 下载并安装
-    - 使用默认安装路径安装:
-      * 64位系统: C:\Program Files\7-Zip
-      * 32位系统: C:\Program Files (x86)\7-Zip
-    - 脚本会自动检测这些路径
+    - 将7-Zip安装目录添加到系统PATH:
+      1. 打开"系统属性" > "环境变量"
+      2. 在"系统变量"中找到"Path"
+      3. 点击"编辑" > "新建"
+      4. 添加7-Zip安装目录(默认为C:\Program Files\7-Zip)
+      5. 点击"确定"保存
+    - 重启终端使PATH生效
 
 6. **下载预训练模型和资源**
     ```cmd
+    git clone https://github.com/Jeffreytsai1004/GaussianHairCut
+    cd GaussianHairCut
     # 在PowerShell中运行:
     # 脚本会自动安装gdown并下载所需资源
     .\download_resource.bat

download_resource.bat

@@ -5,20 +5,12 @@ REM 设置当前目录
 SET PROJECT_DIR=%~dp0
 cd %PROJECT_DIR%
 
-REM 检查7-Zip安装路径
-SET "PROG_FILES=C:\Program Files"
-SET "PROG_FILES_X86=C:\Program Files (x86)"
-
-IF EXIST "%PROG_FILES%\7-Zip\7z.exe" (
-    SET "SEVENZIP=%PROG_FILES%\7-Zip\7z.exe"
-) ELSE IF EXIST "%PROG_FILES_X86%\7-Zip\7z.exe" (
-    SET "SEVENZIP=%PROG_FILES_X86%\7-Zip\7z.exe"
-) ELSE (
+REM 检查7-Zip
+where 7z >nul 2>nul
+IF %ERRORLEVEL% NEQ 0 (
     echo ERROR: 7-Zip not found
     echo Please install 7-Zip from https://7-zip.org/
-    echo Default install locations checked:
-    echo - C:\Program Files\7-Zip\7z.exe
-    echo - C:\Program Files (x86)\7-Zip\7z.exe
+    echo And add it to your PATH environment variable
     exit /b 1
 )
 
@@ -32,7 +24,7 @@ IF %ERRORLEVEL% NEQ 0 (
 
 REM 安装gdown
 echo Installing gdown...
-python -m pip install --user gdown --quiet
+python -m pip install --user gdown requests --quiet --upgrade
 IF %ERRORLEVEL% NEQ 0 (
     echo ERROR: Failed to install gdown
     echo Please check your internet connection and try again
@@ -55,11 +47,12 @@ IF %ERRORLEVEL% NEQ 0 (
 
 REM 创建resource目录结构
 mkdir resource 2>nul
-mkdir resource\NeuralHaircut\diffusion_prior 2>nul
-mkdir resource\NeuralHaircut\PIXIE 2>nul
-mkdir resource\Matte-Anything 2>nul
-mkdir resource\openpose\models 2>nul
-mkdir resource\hyperIQA\pretrained 2>nul
+cd resource
+mkdir NeuralHaircut\diffusion_prior 2>nul
+mkdir NeuralHaircut\PIXIE 2>nul
+mkdir Matte-Anything 2>nul
+mkdir openpose\models 2>nul
+mkdir hyperIQA\pretrained 2>nul
 
 echo.
 echo Starting downloads...
@@ -67,39 +60,40 @@ echo This may take a while depending on your internet connection...
 echo.
 
 REM 下载 Neural Haircut 文件
-cd resource\NeuralHaircut
+cd NeuralHaircut
 echo Downloading Neural Haircut files...
-python -m gdown --folder https://drive.google.com/drive/folders/1TCdJ0CKR3Q6LviovndOkJaKm8S1T9F_8
+python -m gdown --folder "https://drive.google.com/drive/folders/1TCdJ0CKR3Q6LviovndOkJaKm8S1T9F_8" --retry-times 5
 cd diffusion_prior
-python -m gdown 1_9EOUXHayKiGH5nkrayncln3d6m1uV7f
+python -m gdown "1_9EOUXHayKiGH5nkrayncln3d6m1uV7f" --retry-times 5
 cd ..\PIXIE
-python -m gdown 1mPcGu62YPc4MdkT8FFiOCP629xsENHZf
+python -m gdown "1mPcGu62YPc4MdkT8FFiOCP629xsENHZf" --retry-times 5
+
 REM 使用7-Zip解压tar.gz文件
-call %SEVENZIP% x pixie_data.tar.gz -y
-call %SEVENZIP% x pixie_data.tar -y
+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 Downloading Matte-Anything files...
-curl -LO https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth
-curl -LO https://github.com/IDEA-Research/GroundingDINO/releases/download/v0.1.0-alpha/groundingdino_swint_ogc.pth
-gdown 1d97oKuITCeWgai2Tf3iNilt6rMSSYzkW -O model.pth
+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
 
 REM 下载 OpenPose 模型
 cd ..\openpose
 echo Downloading OpenPose models...
-python -m gdown 1Yn03cKKfVOq4qXmgBMQD20UMRRRkd_tV -O models.tar.gz
-call %SEVENZIP% x models.tar.gz -y
-call %SEVENZIP% x models.tar -y -omodels
+python -m gdown "1Yn03cKKfVOq4qXmgBMQD20UMRRRkd_tV" -O models.tar.gz --retry-times 5
+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 Downloading hyperIQA models...
-python -m gdown 1OOUmnbvpGea0LIGpIWEbOyxfWx6UCiiE
+python -m gdown "1OOUmnbvpGea0LIGpIWEbOyxfWx6UCiiE" --retry-times 5
 
 cd ..\..
 echo.