CGNICO/GaussianHaircut
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update download_resource.bat

Browse Source
This commit is contained in:
Jeffreytsai1004 2025-02-15 16:42:39 +08:00
parent 5002da7bf4
commit d1d5d164dc
1 changed files with 30 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

56
download_resource.bat
View File

@ -5,30 +5,33 @@ REM 设置当前目录
SET PROJECT_DIR=%~dp0 SET PROJECT_DIR=%~dp0
cd %PROJECT_DIR% cd %PROJECT_DIR%
REM 检查7-Zip REM 检查7-Zip是否安装
where 7z >nul 2>nul where 7z >nul 2>nul
IF %ERRORLEVEL% NEQ 0 ( IF %ERRORLEVEL% NEQ 0 (
echo ERROR: 7-Zip not found echo 错误：未找到7-Zip
echo Please install 7-Zip from https://7-zip.org/ echo 请从 https://7-zip.org/ 下载并安装7-Zip
echo And add it to your PATH environment variable echo 并将其添加到系统环境变量PATH中
exit /b 1 exit /b 1
) )
REM 检查Python是否安装 REM 检查Python是否安装
python --version >nul 2>nul python --version >nul 2>nul
IF %ERRORLEVEL% NEQ 0 ( IF %ERRORLEVEL% NEQ 0 (
echo ERROR: Python not found echo 错误：未找到Python
echo Please install Python and add it to PATH echo 请安装Python并将其添加到系统环境变量PATH中
exit /b 1 exit /b 1
) )
REM 更新pip
python -m pip install --upgrade pip
REM 安装gdown REM 安装gdown
echo Installing gdown... echo 正在安装gdown...
python -m pip install --user gdown --upgrade python -m pip install --user gdown --upgrade
python -m pip install --user requests --upgrade python -m pip install --user requests --upgrade
IF %ERRORLEVEL% NEQ 0 ( IF %ERRORLEVEL% NEQ 0 (
echo ERROR: Failed to install gdown echo 错误：gdown安装失败
echo Please check your internet connection and try again echo 请检查网络连接后重试
exit /b 1 exit /b 1
) )
@ -39,14 +42,14 @@ set PYTHON_USER_SCRIPTS=%USERPROFILE%\AppData\Roaming\Python\Python3*\Scripts
set PATH=%PYTHON_SCRIPTS_PATH%;%PYTHON_USER_SCRIPTS%;%PATH% set PATH=%PYTHON_SCRIPTS_PATH%;%PYTHON_USER_SCRIPTS%;%PATH%
REM 验证gdown是否可用 REM 验证gdown是否可用
echo Testing gdown installation... echo 正在验证gdown安装...
python -c "import gdown" >nul 2>nul python -c "import gdown" >nul 2>nul
IF %ERRORLEVEL% NEQ 0 ( IF %ERRORLEVEL% NEQ 0 (
echo ERROR: gdown installation failed echo 错误：gdown安装验证失败
echo Please try running the script again echo 请重新运行脚本
exit /b 1 exit /b 1
) )
echo gdown installed successfully! echo gdown安装成功！
echo. echo.
REM 创建resource目录结构 REM 创建resource目录结构
@ -58,14 +61,15 @@ mkdir Matte-Anything 2>nul
mkdir openpose\models 2>nul mkdir openpose\models 2>nul
mkdir hyperIQA\pretrained 2>nul mkdir hyperIQA\pretrained 2>nul
echo. echo ************************************************************
echo Starting downloads... echo * 开始下载所需资源文件 *
echo This may take a while depending on your internet connection... echo * 下载时间取决于您的网络连接速度 *
echo. echo * 请耐心等待... *
echo ************************************************************
REM 下载 Neural Haircut 文件 REM 下载 Neural Haircut 文件
cd NeuralHaircut cd NeuralHaircut
echo Downloading Neural Haircut files... echo 正在下载Neural Haircut文件...
python -m gdown --folder "https://drive.google.com/drive/folders/1TCdJ0CKR3Q6LviovndOkJaKm8S1T9F_8" python -m gdown --folder "https://drive.google.com/drive/folders/1TCdJ0CKR3Q6LviovndOkJaKm8S1T9F_8"
cd diffusion_prior cd diffusion_prior
python -m gdown "1_9EOUXHayKiGH5nkrayncln3d6m1uV7f" python -m gdown "1_9EOUXHayKiGH5nkrayncln3d6m1uV7f"
@ -74,7 +78,7 @@ python -m gdown "1mPcGu62YPc4MdkT8FFiOCP629xsENHZf"
REM 使用7-Zip解压tar.gz文件 REM 使用7-Zip解压tar.gz文件
IF NOT EXIST pixie_data.tar.gz ( IF NOT EXIST pixie_data.tar.gz (
echo ERROR: Failed to download pixie_data.tar.gz echo 错误：下载pixie_data.tar.gz失败
exit /b 1 exit /b 1
) )
7z x pixie_data.tar.gz -y 7z x pixie_data.tar.gz -y
@ -84,14 +88,14 @@ del pixie_data.tar
REM 下载 Matte-Anything 文件 REM 下载 Matte-Anything 文件
cd ..\..\Matte-Anything cd ..\..\Matte-Anything
echo Downloading Matte-Anything files... echo 正在下载Matte-Anything文件...
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 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 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 python -m gdown "1d97oKuITCeWgai2Tf3iNilt6rMSSYzkW" -O model.pth --retry-times 5
REM 下载 OpenPose 模型 REM 下载 OpenPose 模型
cd ..\openpose cd ..\openpose
echo Downloading OpenPose models... echo 正在下载OpenPose模型...
python -m gdown "1Yn03cKKfVOq4qXmgBMQD20UMRRRkd_tV" -O models.tar.gz python -m gdown "1Yn03cKKfVOq4qXmgBMQD20UMRRRkd_tV" -O models.tar.gz
IF NOT EXIST models.tar.gz ( IF NOT EXIST models.tar.gz (
echo ERROR: Failed to download models.tar.gz echo ERROR: Failed to download models.tar.gz
@ -104,18 +108,18 @@ del models.tar
REM 下载 hyperIQA 模型 REM 下载 hyperIQA 模型
cd ..\hyperIQA\pretrained cd ..\hyperIQA\pretrained
echo Downloading hyperIQA models... echo 正在下载hyperIQA模型...
python -m gdown "1OOUmnbvpGea0LIGpIWEbOyxfWx6UCiiE" python -m gdown "1OOUmnbvpGea0LIGpIWEbOyxfWx6UCiiE"
cd ..\.. cd ..\..
echo. echo.
echo Download completed successfully! echo 下载完成！
echo Resource files have been downloaded to the resource directory. echo 资源文件已下载到resource目录。
echo. echo.
IF %ERRORLEVEL% NEQ 0 ( IF %ERRORLEVEL% NEQ 0 (
echo. echo.
echo ERROR: Download failed echo 错误：下载失败
echo Please check your internet connection and try again echo 请检查网络连接后重试
exit /b 1 exit /b 1
) )