Update download_resource.bat

download_resource.bat

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