Update
This commit is contained in:
parent
82b2c68e1e
commit
bbf939c20d
@ -1,47 +1,18 @@
|
||||
@echo off
|
||||
setlocal enabledelayedexpansion
|
||||
|
||||
REM 错误处理
|
||||
:ERROR_HANDLER
|
||||
IF %ERRORLEVEL% NEQ 0 (
|
||||
echo.
|
||||
echo ************************************************************
|
||||
echo * 下载任务被中断 *
|
||||
echo * 请检查网络连接后重新运行脚本 *
|
||||
echo * 如果问题持续存在，请查看以上错误信息 *
|
||||
echo ************************************************************
|
||||
exit /b 1
|
||||
)
|
||||
|
||||
REM 设置当前目录
|
||||
SET PROJECT_DIR=%~dp0
|
||||
cd %PROJECT_DIR%
|
||||
|
||||
REM 创建resource目录结构
|
||||
echo 正在创建resource目录结构...
|
||||
mkdir resource 2>nul
|
||||
mkdir resource\NeuralHaircut 2>nul
|
||||
mkdir resource\NeuralHaircut\PIXIE 2>nul
|
||||
mkdir resource\Matte-Anything 2>nul
|
||||
mkdir resource\openpose 2>nul
|
||||
mkdir resource\hyperIQA 2>nul
|
||||
|
||||
REM 验证已有文件
|
||||
echo 正在检查已有资源文件...
|
||||
SET MISSING_FILES=0
|
||||
SET /A TOTAL_FILES=7
|
||||
SET /A EXISTING_FILES=0
|
||||
SET MISSING_LIST=
|
||||
|
||||
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
|
||||
)
|
||||
|
||||
echo ======================================= 基础环境配置 ==============================================
|
||||
echo.
|
||||
echo **************************************************************************************************
|
||||
echo * 基础环境配置 *
|
||||
echo * 检查Python是否安装，更新pip，安装gdown *
|
||||
echo * 请耐心等待... *
|
||||
echo **************************************************************************************************
|
||||
echo.
|
||||
REM 检查Python是否安装
|
||||
python --version >nul 2>nul
|
||||
IF %ERRORLEVEL% NEQ 0 (
|
||||
@ -75,6 +46,16 @@ 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
|
||||
@ -86,50 +67,57 @@ IF %ERRORLEVEL% NEQ 0 (
|
||||
echo gdown安装成功！
|
||||
echo.
|
||||
|
||||
REM Download Neural Haircut files
|
||||
echo ======================================= 开始下载所需资源文件 =======================================
|
||||
echo.
|
||||
echo ************************************************************
|
||||
echo * 开始下载所需资源文件 *
|
||||
echo * 下载时间取决于您的网络连接速度 *
|
||||
echo * 请耐心等待... *
|
||||
echo ************************************************************
|
||||
echo **************************************************************************************************
|
||||
echo * 开始下载所需资源文件 *
|
||||
echo * 下载时间取决于您的网络连接速度 *
|
||||
echo * 请耐心等待... *
|
||||
echo **************************************************************************************************
|
||||
echo.
|
||||
|
||||
echo [1/7] 正在下载Neural Haircut文件...
|
||||
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%
|
||||
|
||||
REM 更新 diffusion_prior 模型
|
||||
cd %PROJECT_DIR%\resource\NeuralHaircut\pretrained_models\diffusion_prior
|
||||
echo 正在下载Neural Haircut扩散先验模型文件...
|
||||
echo [2/8] 正在下载Neural Haircut扩散先验模型文件...
|
||||
mkdir %PROJECT_DIR%/resource/NeuralHaircut/pretrained_models/diffusion_prior
|
||||
cd %PROJECT_DIR%/resource/NeuralHaircut/pretrained_models/diffusion_prior
|
||||
python -m gdown "1_9EOUXHayKiGH5nkrayncln3d6m1uV7f"
|
||||
cd %PROJECT_DIR%
|
||||
|
||||
REM 下载 PIXIE 文件
|
||||
echo [3/8] 正在下载Neural HaircutPIXIE模型文件...
|
||||
mkdir %PROJECT_DIR%/resource/NeuralHaircut/PIXIE
|
||||
cd %PROJECT_DIR%/resource/NeuralHaircut/PIXIE
|
||||
echo 正在下载Neural HaircutPIXIE模型文件...
|
||||
python -m gdown "1mPcGu62YPc4MdkT8FFiOCP629xsENHZf" && tar -xvzf pixie_data.tar.gz ./ && rm pixie_data.tar.gz
|
||||
cd %PROJECT_DIR%
|
||||
|
||||
REM 下载 hyperIQA 模型
|
||||
echo [4/8] 正在下载hyperIQA模型文件...
|
||||
mkdir %PROJECT_DIR%/resource/hyperIQA/pretrained
|
||||
cd %PROJECT_DIR%/resource/hyperIQA/pretrained
|
||||
echo 正在下载hyperIQA模型文件...
|
||||
python -m gdown "1OOUmnbvpGea0LIGpIWEbOyxfWx6UCiiE"
|
||||
cd %PROJECT_DIR%
|
||||
|
||||
REM Download Matte-Anything files
|
||||
echo [5/8] 正在下载Matte-Anything扩散先验模型文件...
|
||||
mkdir %PROJECT_DIR%/resource/Matte-Anything/pretrained
|
||||
cd %PROJECT_DIR%/resource/Matte-Anything/pretrained
|
||||
echo 正在下载Matte-Anything扩散先验模型文件...
|
||||
wget https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth
|
||||
echo 正在下载Matte-AnythingGroundingDINO模型文件...
|
||||
wget https://github.com/IDEA-Research/GroundingDINO/releases/download/v0.1.0-alpha/groundingdino_swint_ogc.pth
|
||||
echo 正在下载Matte-Anything模型文件...
|
||||
wget "https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth"
|
||||
echo [6/8] 正在下载Matte-AnythingGroundingDINO模型文件...
|
||||
mkdir %PROJECT_DIR%/resource/Matte-Anything/GroundingDINO
|
||||
cd %PROJECT_DIR%/resource/Matte-Anything/GroundingDINO
|
||||
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"
|
||||
|
||||
REM Download OpenPose files
|
||||
cd %PROJECT_DIR%/resource/openpose
|
||||
echo 正在下载OpenPose模型文件...
|
||||
python -m gdown "1Yn03cKKfVOq4qXmgBMQD20UMRRRkd_tV" && tar -xvzf models.tar.gz && rm models.tar.gz
|
||||
|
||||
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 所有模型文件下载完成！
|
||||
|
||||
|
||||
|
||||
82
install.bat
82
install.bat
@ -165,72 +165,14 @@ IF NOT EXIST "%PROJECT_DIR%\resource" (
|
||||
echo resource/
|
||||
echo ├── NeuralHaircut/
|
||||
echo │ ├── diffusion_prior/
|
||||
echo │ │ └── model.pt
|
||||
echo │ │ └── dif_ckpt.pt
|
||||
echo │ ├── strand_prior/
|
||||
echo │ │ └── model.pt
|
||||
echo │ │ └── strand_ckpt.pt
|
||||
echo │ └── PIXIE/
|
||||
echo │ └── pixie_data/
|
||||
echo ├── Matte-Anything/
|
||||
echo │ ├── sam_vit_h_4b8939.pth
|
||||
echo │ ├── groundingdino_swint_ogc.pth
|
||||
echo │ └── model.pth
|
||||
echo ├── openpose/
|
||||
echo │ └── models/
|
||||
echo │ └── pose/
|
||||
echo │ └── coco/
|
||||
echo │ └── pose_iter_584000.caffemodel
|
||||
echo └── hyperIQA/
|
||||
echo └── pretrained/
|
||||
echo └── hyperIQA.pth
|
||||
exit /b 1
|
||||
)
|
||||
|
||||
REM 检查模型文件是否存在
|
||||
IF NOT EXIST "%PROJECT_DIR%\resource\NeuralHaircut\diffusion_prior\model.pt" (
|
||||
echo ERROR: Neural Haircut Diffusion Prior model not found
|
||||
echo Please run download_resource.bat first
|
||||
exit /b 1
|
||||
)
|
||||
|
||||
IF NOT EXIST "%PROJECT_DIR%\resource\NeuralHaircut\strand_prior\model.pt" (
|
||||
echo ERROR: Neural Haircut Strand Prior model not found
|
||||
echo Please run download_resource.bat first
|
||||
exit /b 1
|
||||
)
|
||||
|
||||
IF NOT EXIST "%PROJECT_DIR%\resource\NeuralHaircut\PIXIE\pixie_data\pixie_data" (
|
||||
echo ERROR: PIXIE model not found
|
||||
echo Please run download_resource.bat first
|
||||
exit /b 1
|
||||
)
|
||||
|
||||
IF NOT EXIST "%PROJECT_DIR%\resource\Matte-Anything\sam_vit_h_4b8939.pth" (
|
||||
echo ERROR: SAM model not found
|
||||
echo Please run download_resource.bat first
|
||||
exit /b 1
|
||||
)
|
||||
|
||||
IF NOT EXIST "%PROJECT_DIR%\resource\Matte-Anything\groundingdino_swint_ogc.pth" (
|
||||
echo ERROR: GroundingDINO model not found
|
||||
echo Please run download_resource.bat first
|
||||
exit /b 1
|
||||
)
|
||||
|
||||
IF NOT EXIST "%PROJECT_DIR%\resource\Matte-Anything\model.pth" (
|
||||
echo ERROR: Matte-Anything model not found
|
||||
echo Please run download_resource.bat first
|
||||
exit /b 1
|
||||
)
|
||||
|
||||
IF NOT EXIST "%PROJECT_DIR%\resource\openpose\models\pose\coco\pose_iter_584000.caffemodel" (
|
||||
echo ERROR: OpenPose model not found
|
||||
echo Please run download_resource.bat first
|
||||
exit /b 1
|
||||
)
|
||||
|
||||
IF NOT EXIST "%PROJECT_DIR%\resource\hyperIQA\pretrained\hyperIQA.pth" (
|
||||
echo ERROR: hyperIQA model not found
|
||||
echo Please run download_resource.bat first
|
||||
exit /b 1
|
||||
)
|
||||
|
||||
@ -247,23 +189,11 @@ mkdir "%PROJECT_DIR%\ext\Matte-Anything\pretrained" 2>nul
|
||||
mkdir "%PROJECT_DIR%\ext\openpose\models\pose\coco" 2>nul
|
||||
mkdir "%PROJECT_DIR%\ext\hyperIQA\pretrained" 2>nul
|
||||
|
||||
REM Neural Haircut
|
||||
xcopy /E /I /Y "%PROJECT_DIR%\resource\NeuralHaircut\*" "%PROJECT_DIR%\ext\NeuralHaircut\"
|
||||
|
||||
REM PIXIE
|
||||
cd "%PROJECT_DIR%\ext\PIXIE"
|
||||
xcopy /E /I /Y "%PROJECT_DIR%\resource\NeuralHaircut\PIXIE\*" .
|
||||
|
||||
REM Matte-Anything
|
||||
xcopy /Y "%PROJECT_DIR%\resource\Matte-Anything\sam_vit_h_4b8939.pth" "%PROJECT_DIR%\ext\Matte-Anything\pretrained\"
|
||||
xcopy /Y "%PROJECT_DIR%\resource\Matte-Anything\groundingdino_swint_ogc.pth" "%PROJECT_DIR%\ext\Matte-Anything\pretrained\"
|
||||
xcopy /Y "%PROJECT_DIR%\resource\Matte-Anything\model.pth" "%PROJECT_DIR%\ext\Matte-Anything\"
|
||||
|
||||
REM OpenPose
|
||||
xcopy /Y "%PROJECT_DIR%\resource\openpose\models\pose_iter_584000.caffemodel" "%PROJECT_DIR%\ext\openpose\models\pose\coco\"
|
||||
|
||||
REM hyperIQA
|
||||
xcopy /Y "%PROJECT_DIR%\resource\hyperIQA\pretrained\hyperIQA.pth" "%PROJECT_DIR%\ext\hyperIQA\pretrained\"
|
||||
xcopy /E /I /Y "%PROJECT_DIR%\resource\NeuralHaircut\PIXIE\*" "%PROJECT_DIR%\ext\PIXIE"
|
||||
xcopy /Y "%PROJECT_DIR%\resource\Matte-Anything\*" "%PROJECT_DIR%\ext\Matte-Anything\pretrained\"
|
||||
xcopy /Y "%PROJECT_DIR%\resource\openpose\*" "%PROJECT_DIR%\ext\openpose\*"
|
||||
xcopy /Y "%PROJECT_DIR%\resource\hyperIQA\*" "%PROJECT_DIR%\ext\hyperIQA\*"
|
||||
|
||||
REM 安装 PIXIE 环境
|
||||
CALL activate_pixie-env.bat
|
||||
|
||||
Loading…
Reference in New Issue
Block a user