Update

README.md

@@ -96,13 +96,18 @@
 
 5. **安装 7-Zip**
     - 从 https://7-zip.org/ 下载并安装
-    - 使用默认安装路径安装:
+    - 将7-Zip安装目录添加到系统PATH:
-      * 64位系统: C:\Program Files\7-Zip
+      1. 打开"系统属性" > "环境变量"
-      * 32位系统: C:\Program Files (x86)\7-Zip
+      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,17 +5,12 @@ REM 设置当前目录
 SET PROJECT_DIR=%~dp0
 cd %PROJECT_DIR%
 
-REM 检查7-Zip安装路径
+REM 检查7-Zip
-SET "SEVENZIP_PATH=C:\Program Files\7-Zip\7z.exe"
+where 7z >nul 2>nul
-IF NOT EXIST "%SEVENZIP_PATH%" (
+IF %ERRORLEVEL% NEQ 0 (
-    SET "SEVENZIP_PATH=C:\Program Files (x86)\7-Zip\7z.exe"
-)
-IF NOT EXIST "%SEVENZIP_PATH%" (
     echo ERROR: 7-Zip not found
     echo Please install 7-Zip from https://7-zip.org/
-    echo Default install locations checked:
+    echo And add it to your PATH environment variable
-    echo - C:\Program Files\7-Zip\7z.exe
-    echo - C:\Program Files (x86)\7-Zip\7z.exe
     exit /b 1
 )
 
@@ -29,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
@@ -52,11 +47,12 @@ IF %ERRORLEVEL% NEQ 0 (
 
 REM 创建resource目录结构
 mkdir resource 2>nul
-mkdir resource\NeuralHaircut\diffusion_prior 2>nul
+cd resource
-mkdir resource\NeuralHaircut\PIXIE 2>nul
+mkdir NeuralHaircut\diffusion_prior 2>nul
-mkdir resource\Matte-Anything 2>nul
+mkdir NeuralHaircut\PIXIE 2>nul
-mkdir resource\openpose\models 2>nul
+mkdir Matte-Anything 2>nul
-mkdir resource\hyperIQA\pretrained 2>nul
+mkdir openpose\models 2>nul
+mkdir hyperIQA\pretrained 2>nul
 
 echo.
 echo Starting downloads...
@@ -64,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文件
-"%SEVENZIP_PATH%" x pixie_data.tar.gz -y
+7z x pixie_data.tar.gz -y
-"%SEVENZIP_PATH%" x pixie_data.tar -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 -L --retry 5 -o sam_vit_h_4b8939.pth 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
+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
-gdown 1d97oKuITCeWgai2Tf3iNilt6rMSSYzkW -O model.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
+python -m gdown "1Yn03cKKfVOq4qXmgBMQD20UMRRRkd_tV" -O models.tar.gz --retry-times 5
-"%SEVENZIP_PATH%" x models.tar.gz -y
+7z x models.tar.gz -y
-"%SEVENZIP_PATH%" x models.tar -y -omodels
+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.