diff --git a/README.md b/README.md index 61e7f37..e0d7f78 100644 --- a/README.md +++ b/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 diff --git a/download_resource.bat b/download_resource.bat index 3136403..5e0c434 100644 --- a/download_resource.bat +++ b/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.