142 lines
4.7 KiB
Batchfile
142 lines
4.7 KiB
Batchfile
@echo off
|
||
setlocal enabledelayedexpansion
|
||
|
||
SET PROJECT_DIR=%~dp0
|
||
cd %PROJECT_DIR%
|
||
|
||
|
||
echo ======================================= 基础环境配置 ==============================================
|
||
echo.
|
||
echo **************************************************************************************************
|
||
echo * 基础环境配置 *
|
||
echo * 检查Python是否安装，更新pip，安装gdown *
|
||
echo * 请耐心等待... *
|
||
echo **************************************************************************************************
|
||
echo.
|
||
REM 检查Python是否安装
|
||
python --version >nul 2>nul
|
||
IF %ERRORLEVEL% NEQ 0 (
|
||
echo 错误：未找到Python
|
||
echo 请安装Python并将其添加到系统环境变量PATH中
|
||
exit /b 1
|
||
)
|
||
|
||
REM 更新pip
|
||
python -m pip install --upgrade pip
|
||
|
||
REM 检查gdown是否已安装
|
||
python -c "import gdown" >nul 2>nul
|
||
IF %ERRORLEVEL% EQU 0 (
|
||
echo gdown已安装，跳过安装步骤...
|
||
) ELSE (
|
||
REM 安装gdown
|
||
echo 正在安装gdown...
|
||
python -m pip install --user gdown --upgrade
|
||
python -m pip install --user requests --upgrade
|
||
IF %ERRORLEVEL% NEQ 0 (
|
||
echo 错误：gdown安装失败
|
||
echo 请检查网络连接后重试
|
||
exit /b 1
|
||
)
|
||
)
|
||
|
||
REM 添加Python Scripts目录到PATH
|
||
for /f "tokens=*" %%i in ('python -c "import sys; print(sys.executable)"') do set PYTHON_PATH=%%i
|
||
set PYTHON_SCRIPTS_PATH=%PYTHON_PATH:python.exe=Scripts%
|
||
set PYTHON_USER_SCRIPTS=%USERPROFILE%\AppData\Roaming\Python\Python3*\Scripts
|
||
set PATH=%PYTHON_SCRIPTS_PATH%;%PYTHON_USER_SCRIPTS%;%PATH%
|
||
|
||
REM 检查7-Zip是否安装
|
||
where 7z >nul 2>nul
|
||
IF %ERRORLEVEL% NEQ 0 (
|
||
echo 错误：未找到7-Zip
|
||
echo 请从 https://7-zip.org/ 下载并安装7-Zip
|
||
echo 并将其添加到系统环境变量PATH中
|
||
exit /b 1
|
||
)
|
||
|
||
|
||
REM 验证gdown是否可用
|
||
echo 正在验证gdown安装...
|
||
python -c "import gdown" >nul 2>nul
|
||
IF %ERRORLEVEL% NEQ 0 (
|
||
echo 错误：gdown安装验证失败
|
||
echo 请重新运行脚本
|
||
exit /b 1
|
||
)
|
||
echo gdown安装成功！
|
||
echo.
|
||
|
||
echo ======================================= 开始下载所需资源文件 =======================================
|
||
echo.
|
||
echo **************************************************************************************************
|
||
echo * 开始下载所需资源文件 *
|
||
echo * 下载时间取决于您的网络连接速度 *
|
||
echo * 请耐心等待... *
|
||
echo **************************************************************************************************
|
||
echo.
|
||
|
||
echo [1/8] 正在下载Neural Haircut文件...
|
||
mkdir %PROJECT_DIR%/resource/NeuralHaircut
|
||
cd %PROJECT_DIR%/resource/NeuralHaircut
|
||
python -m gdown --folder "https://drive.google.com/drive/folders/1TCdJ0CKR3Q6LviovndOkJaKm8S1T9F_8"
|
||
cd %PROJECT_DIR%
|
||
|
||
echo [2/8] 正在下载Neural Haircut扩散先验模型文件...
|
||
cd %PROJECT_DIR%/resource/NeuralHaircut/diffusion_prior
|
||
python -m gdown "1_9EOUXHayKiGH5nkrayncln3d6m1uV7f"
|
||
cd %PROJECT_DIR%
|
||
|
||
echo [3/8] 正在下载Neural HaircutPIXIE模型文件...
|
||
mkdir %PROJECT_DIR%/resource/NeuralHaircut/PIXIE
|
||
cd %PROJECT_DIR%/resource/NeuralHaircut/PIXIE
|
||
python -m gdown "1mPcGu62YPc4MdkT8FFiOCP629xsENHZf" && tar -xvzf pixie_data.tar.gz ./ && rm pixie_data.tar.gz
|
||
cd %PROJECT_DIR%
|
||
|
||
echo [4/8] 正在下载hyperIQA模型文件...
|
||
mkdir %PROJECT_DIR%/resource/hyperIQA/pretrained
|
||
cd %PROJECT_DIR%/resource/hyperIQA/pretrained
|
||
python -m gdown "1OOUmnbvpGea0LIGpIWEbOyxfWx6UCiiE"
|
||
cd %PROJECT_DIR%
|
||
|
||
echo [5/8] 正在下载Matte-Anything扩散先验模型文件...
|
||
mkdir %PROJECT_DIR%/resource/Matte-Anything/pretrained
|
||
cd %PROJECT_DIR%/resource/Matte-Anything/pretrained
|
||
wget "https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth"
|
||
echo [6/8] 正在下载Matte-AnythingGroundingDINO模型文件...
|
||
cd %PROJECT_DIR%/resource/Matte-Anything/pretrained
|
||
wget "https://github.com/IDEA-Research/GroundingDINO/releases/download/v0.1.0-alpha/groundingdino_swint_ogc.pth"
|
||
echo [7/8] 正在下载Matte-Anything模型文件...
|
||
python -m gdown "1d97oKuITCeWgai2Tf3iNilt6rMSSYzkW"
|
||
cd %PROJECT_DIR%
|
||
|
||
echo [8/8] 正在下载OpenPose模型文件...
|
||
mkdir %PROJECT_DIR%/resource/openpose/models/pose/coco
|
||
cd %PROJECT_DIR%/resource/openpose/models/pose/coco
|
||
python -m gdown "1Yn03cKKfVOq4qXmgBMQD20UMRRRkd_tV" && tar -xvzf models.tar.gz && rm models.tar.gz
|
||
cd %PROJECT_DIR%
|
||
|
||
echo 所有模型文件下载完成！
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|