diff --git a/download_resource.bat b/download_resource.bat index 07cc4db..49a3447 100644 --- a/download_resource.bat +++ b/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 )