Update download_resource.bat

This commit is contained in:
Jeffreytsai1004 2025-02-16 17:04:07 +08:00
parent 0a7148eb40
commit e19a213d21

View File

@ -163,22 +163,22 @@ IF EXIST "%DINO_PATH%" (
REM OpenPose 模型检查 REM OpenPose 模型检查
IF EXIST "%OPENPOSE_PATH%" ( IF EXIST "%OPENPOSE_PATH%" (
echo [√] %MODEL_OPENPOSE_NAME%已存在 echo [√] OpenPose模型已存在
SET /A EXISTING_FILES+=1 SET /A EXISTING_FILES+=1
) ELSE ( ) ELSE (
echo [×] 缺少%MODEL_OPENPOSE_NAME% echo [×] 缺少OpenPose模型
SET /A MISSING_FILES+=1 SET /A MISSING_FILES+=1
SET MISSING_LIST=!MISSING_LIST!%MODEL_OPENPOSE_NAME%, SET MISSING_LIST=!MISSING_LIST!OpenPose模型,
) )
REM hyperIQA 模型检查 REM hyperIQA 模型检查
IF EXIST "%HYPERIQA_PATH%" ( IF EXIST "%HYPERIQA_PATH%" (
echo [√] %MODEL_HYPERIQA_NAME%已存在 echo [√] hyperIQA模型已存在
SET /A EXISTING_FILES+=1 SET /A EXISTING_FILES+=1
) ELSE ( ) ELSE (
echo [×] 缺少%MODEL_HYPERIQA_NAME% echo [×] 缺少hyperIQA模型
SET /A MISSING_FILES+=1 SET /A MISSING_FILES+=1
SET MISSING_LIST=!MISSING_LIST!%MODEL_HYPERIQA_NAME%, SET MISSING_LIST=!MISSING_LIST!hyperIQA模型,
) )
echo. echo.
@ -270,11 +270,40 @@ IF NOT EXIST "%NEURAL_HAIRCUT_FOLDER_PATH%\pretrained_models\strand_prior\strand
echo [√] Neural Haircut Strand Prior验证成功 echo [√] Neural Haircut Strand Prior验证成功
) )
IF NOT EXIST "%PIXIE_PATH%" echo 警告缺少PIXIE模型 IF NOT EXIST "%PIXIE_PATH%" (
IF NOT EXIST "%SAM_PATH%" echo 警告缺少SAM模型 echo [×] 缺少PIXIE模型
IF NOT EXIST "%DINO_PATH%" echo 警告缺少GroundingDINO模型 SET /A VERIFY_FAILED+=1
IF NOT EXIST "%OPENPOSE_PATH%" echo 警告缺少OpenPose模型 ) ELSE (
IF NOT EXIST "%HYPERIQA_PATH%" echo 警告缺少hyperIQA模型 echo [√] PIXIE模型验证成功
)
IF NOT EXIST "%SAM_PATH%" (
echo [×] 缺少SAM模型
SET /A VERIFY_FAILED+=1
) ELSE (
echo [√] SAM模型验证成功
)
IF NOT EXIST "%DINO_PATH%" (
echo [×] 缺少GroundingDINO模型
SET /A VERIFY_FAILED+=1
) ELSE (
echo [√] GroundingDINO模型验证成功
)
IF NOT EXIST "%OPENPOSE_PATH%" (
echo [×] 缺少OpenPose模型
SET /A VERIFY_FAILED+=1
) ELSE (
echo [√] OpenPose模型验证成功
)
IF NOT EXIST "%HYPERIQA_PATH%" (
echo [×] 缺少hyperIQA模型
SET /A VERIFY_FAILED+=1
) ELSE (
echo [√] hyperIQA模型验证成功
)
IF %VERIFY_FAILED% GTR 0 ( IF %VERIFY_FAILED% GTR 0 (
echo. echo.