README.md

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

download_resource.bat

@@ -5,12 +5,17 @@ REM 设置当前目录
 SET PROJECT_DIR=%~dp0
 cd %PROJECT_DIR%
 
-REM 检查7-Zip
-where 7z >nul 2>nul
-IF %ERRORLEVEL% NEQ 0 (
+REM 检查7-Zip安装路径
+SET "SEVENZIP_PATH=C:\Program Files\7-Zip\7z.exe"
+IF NOT EXIST "%SEVENZIP_PATH%" (
+    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 And add it to your PATH environment variable
+    echo Default install locations checked:
+    echo - C:\Program Files\7-Zip\7z.exe
+    echo - C:\Program Files (x86)\7-Zip\7z.exe
     exit /b 1
 )
 
@@ -24,7 +29,7 @@ IF %ERRORLEVEL% NEQ 0 (
 
 REM 安装gdown
 echo Installing gdown...
-python -m pip install --user gdown requests --quiet --upgrade
+python -m pip install --user gdown --quiet
 IF %ERRORLEVEL% NEQ 0 (
     echo ERROR: Failed to install gdown
     echo Please check your internet connection and try again
@@ -47,12 +52,11 @@ IF %ERRORLEVEL% NEQ 0 (
 
 REM 创建resource目录结构
 mkdir resource 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
+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
 
 echo.
 echo Starting downloads...
@@ -60,40 +64,39 @@ echo This may take a while depending on your internet connection...
 echo.
 
 REM 下载 Neural Haircut 文件
-cd NeuralHaircut
+cd resource\NeuralHaircut
 echo Downloading Neural Haircut files...
-python -m gdown --folder "https://drive.google.com/drive/folders/1TCdJ0CKR3Q6LviovndOkJaKm8S1T9F_8" --retry-times 5
+python -m gdown --folder https://drive.google.com/drive/folders/1TCdJ0CKR3Q6LviovndOkJaKm8S1T9F_8
 cd diffusion_prior
-python -m gdown "1_9EOUXHayKiGH5nkrayncln3d6m1uV7f" --retry-times 5
+python -m gdown 1_9EOUXHayKiGH5nkrayncln3d6m1uV7f
 cd ..\PIXIE
-python -m gdown "1mPcGu62YPc4MdkT8FFiOCP629xsENHZf" --retry-times 5
-
+python -m gdown 1mPcGu62YPc4MdkT8FFiOCP629xsENHZf
 REM 使用7-Zip解压tar.gz文件
-7z x pixie_data.tar.gz -y
-7z x pixie_data.tar -y
+"%SEVENZIP_PATH%" x pixie_data.tar.gz -y
+"%SEVENZIP_PATH%" 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 -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
+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
 
 REM 下载 OpenPose 模型
 cd ..\openpose
 echo Downloading OpenPose models...
-python -m gdown "1Yn03cKKfVOq4qXmgBMQD20UMRRRkd_tV" -O models.tar.gz --retry-times 5
-7z x models.tar.gz -y
-7z x models.tar -y -omodels
+python -m gdown 1Yn03cKKfVOq4qXmgBMQD20UMRRRkd_tV -O models.tar.gz
+"%SEVENZIP_PATH%" x models.tar.gz -y
+"%SEVENZIP_PATH%" 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" --retry-times 5
+python -m gdown 1OOUmnbvpGea0LIGpIWEbOyxfWx6UCiiE
 
 cd ..\..
 echo.