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:36:21 +08:00
parent 51cf2252da
commit 5002da7bf4
1 changed files with 21 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
download_resource.bat
View File

@ -24,7 +24,8 @@ IF %ERRORLEVEL% NEQ 0 (
REM 安装gdown
echo Installing gdown...
python -m pip install --user gdown requests --quiet --upgrade
python -m pip install --user gdown --upgrade
python -m pip install --user requests --upgrade
IF %ERRORLEVEL% NEQ 0 (
echo ERROR: Failed to install gdown
echo Please check your internet connection and try again
@ -38,12 +39,15 @@ set PYTHON_USER_SCRIPTS=%USERPROFILE%\AppData\Roaming\Python\Python3*\Scripts
set PATH=%PYTHON_SCRIPTS_PATH%;%PYTHON_USER_SCRIPTS%;%PATH%
REM 验证gdown是否可用
python -m gdown --version >nul 2>nul
echo Testing gdown installation...
python -c "import gdown" >nul 2>nul
IF %ERRORLEVEL% NEQ 0 (
echo ERROR: gdown installation found but not accessible
echo Please try reinstalling gdown: python -m pip install --user --upgrade gdown
echo ERROR: gdown installation failed
echo Please try running the script again
exit /b 1
)
echo gdown installed successfully!
echo.
REM 创建resource目录结构
mkdir resource 2>nul
@ -62,13 +66,17 @@ echo.
REM 下载 Neural Haircut 文件
cd 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文件
IF NOT EXIST pixie_data.tar.gz (
echo ERROR: Failed to download pixie_data.tar.gz
exit /b 1
)
7z x pixie_data.tar.gz -y
7z x pixie_data.tar -y
del pixie_data.tar.gz
@ -84,7 +92,11 @@ 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 --retry-times 5
python -m gdown "1Yn03cKKfVOq4qXmgBMQD20UMRRRkd_tV" -O models.tar.gz
IF NOT EXIST models.tar.gz (
echo ERROR: Failed to download models.tar.gz
exit /b 1
)
7z x models.tar.gz -y
7z x models.tar -y -omodels
del models.tar.gz
@ -93,7 +105,7 @@ 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.