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-16 14:50:36 +08:00
parent 5c81a20afc
commit 94e9843c08
1 changed files with 27 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
download_resource.bat
View File

@ -72,59 +72,79 @@ echo 正在检查已有资源文件...
SET MISSING_FILES=0 SET MISSING_FILES=0
SET /A TOTAL_FILES=6 SET /A TOTAL_FILES=6
SET /A EXISTING_FILES=0 SET /A EXISTING_FILES=0
SET MISSING_LIST=
IF EXIST "NeuralHaircut\diffusion_prior\model.pt" ( REM Neural Haircut 模型检查
SET MODEL_PATH=NeuralHaircut\diffusion_prior\model.pt
IF EXIST "%MODEL_PATH%" (
echo [√] Neural Haircut模型已存在 echo [√] Neural Haircut模型已存在
SET /A EXISTING_FILES+=1 SET /A EXISTING_FILES+=1
) ELSE ( ) ELSE (
echo [×] 缺少Neural Haircut模型 echo [×] 缺少Neural Haircut模型
SET /A MISSING_FILES+=1 SET /A MISSING_FILES+=1
SET MISSING_LIST=!MISSING_LIST!Neural Haircut模型,
) )
IF EXIST "NeuralHaircut\PIXIE\pixie_data" ( REM PIXIE 模型检查
SET MODEL_PATH=NeuralHaircut\PIXIE\pixie_data\data
IF EXIST "%MODEL_PATH%" (
echo [√] PIXIE模型数据已存在 echo [√] PIXIE模型数据已存在
SET /A EXISTING_FILES+=1 SET /A EXISTING_FILES+=1
) ELSE ( ) ELSE (
echo [×] 缺少PIXIE模型数据 echo [×] 缺少PIXIE模型数据
SET /A MISSING_FILES+=1 SET /A MISSING_FILES+=1
SET MISSING_LIST=!MISSING_LIST!PIXIE模型数据,
) )
IF EXIST "Matte-Anything\sam_vit_h_4b8939.pth" ( REM SAM 模型检查
SET MODEL_PATH=Matte-Anything\sam_vit_h_4b8939.pth
IF EXIST "%MODEL_PATH%" (
echo [√] SAM模型已存在 echo [√] SAM模型已存在
SET /A EXISTING_FILES+=1 SET /A EXISTING_FILES+=1
) ELSE ( ) ELSE (
echo [×] 缺少SAM模型 echo [×] 缺少SAM模型
SET /A MISSING_FILES+=1 SET /A MISSING_FILES+=1
SET MISSING_LIST=!MISSING_LIST!SAM模型,
) )
IF EXIST "Matte-Anything\groundingdino_swint_ogc.pth" ( REM GroundingDINO 模型检查
SET MODEL_PATH=Matte-Anything\groundingdino_swint_ogc.pth
IF EXIST "%MODEL_PATH%" (
echo [√] GroundingDINO模型已存在 echo [√] GroundingDINO模型已存在
SET /A EXISTING_FILES+=1 SET /A EXISTING_FILES+=1
) ELSE ( ) ELSE (
echo [×] 缺少GroundingDINO模型 echo [×] 缺少GroundingDINO模型
SET /A MISSING_FILES+=1 SET /A MISSING_FILES+=1
SET MISSING_LIST=!MISSING_LIST!GroundingDINO模型,
) )
IF EXIST "openpose\models\pose\coco" ( REM OpenPose 模型检查
SET MODEL_PATH=openpose\models\pose\coco\pose_iter_584000.caffemodel
IF EXIST "%MODEL_PATH%" (
echo [√] OpenPose模型已存在 echo [√] OpenPose模型已存在
SET /A EXISTING_FILES+=1 SET /A EXISTING_FILES+=1
) ELSE ( ) ELSE (
echo [×] 缺少OpenPose模型 echo [×] 缺少OpenPose模型
SET /A MISSING_FILES+=1 SET /A MISSING_FILES+=1
SET MISSING_LIST=!MISSING_LIST!OpenPose模型,
) )
IF EXIST "hyperIQA\pretrained\hyperIQA.pth" ( REM hyperIQA 模型检查
SET MODEL_PATH=hyperIQA\pretrained\hyperIQA.pth
IF EXIST "%MODEL_PATH%" (
echo [√] hyperIQA模型已存在 echo [√] hyperIQA模型已存在
SET /A EXISTING_FILES+=1 SET /A EXISTING_FILES+=1
) ELSE ( ) ELSE (
echo [×] 缺少hyperIQA模型 echo [×] 缺少hyperIQA模型
SET /A MISSING_FILES+=1 SET /A MISSING_FILES+=1
SET MISSING_LIST=!MISSING_LIST!hyperIQA模型,
) )
echo. echo.
echo 已存在 %EXISTING_FILES%/%TOTAL_FILES% 个模型文件 echo 已存在 %EXISTING_FILES%/%TOTAL_FILES% 个模型文件
IF %MISSING_FILES% GTR 0 ( IF %MISSING_FILES% GTR 0 (
echo 需要下载 %MISSING_FILES% 个模型文件 echo 需要下载 %MISSING_FILES% 个模型文件
echo 缺少的模型: !MISSING_LIST:~0,-2!
echo. echo.
echo ************************************************************ echo ************************************************************
echo * 开始下载所需资源文件 * echo * 开始下载所需资源文件 *
@ -214,7 +234,7 @@ IF EXIST "model.pth" (
REM 下载 OpenPose 模型 REM 下载 OpenPose 模型
cd ..\openpose cd ..\openpose
echo 正在下载OpenPose模型... echo 正在下载OpenPose模型...
IF EXIST "models\pose_iter_584000.caffemodel" ( IF EXIST "models\pose\coco\pose_iter_584000.caffemodel" (
echo OpenPose模型已存在，跳过下载... echo OpenPose模型已存在，跳过下载...
) ELSE ( ) ELSE (
echo 正在下载OpenPose模型... echo 正在下载OpenPose模型...