diff --git a/download_resource.bat b/download_resource.bat index fb2fe98..8d6bab0 100644 --- a/download_resource.bat +++ b/download_resource.bat @@ -163,22 +163,22 @@ IF EXIST "%DINO_PATH%" ( REM OpenPose 模型检查 IF EXIST "%OPENPOSE_PATH%" ( - echo [√] %MODEL_OPENPOSE_NAME%已存在 + echo [√] OpenPose模型已存在 SET /A EXISTING_FILES+=1 ) ELSE ( - echo [×] 缺少%MODEL_OPENPOSE_NAME% + echo [×] 缺少OpenPose模型 SET /A MISSING_FILES+=1 - SET MISSING_LIST=!MISSING_LIST!%MODEL_OPENPOSE_NAME%, + SET MISSING_LIST=!MISSING_LIST!OpenPose模型, ) REM hyperIQA 模型检查 IF EXIST "%HYPERIQA_PATH%" ( - echo [√] %MODEL_HYPERIQA_NAME%已存在 + echo [√] hyperIQA模型已存在 SET /A EXISTING_FILES+=1 ) ELSE ( - echo [×] 缺少%MODEL_HYPERIQA_NAME% + echo [×] 缺少hyperIQA模型 SET /A MISSING_FILES+=1 - SET MISSING_LIST=!MISSING_LIST!%MODEL_HYPERIQA_NAME%, + SET MISSING_LIST=!MISSING_LIST!hyperIQA模型, ) echo. @@ -270,11 +270,40 @@ IF NOT EXIST "%NEURAL_HAIRCUT_FOLDER_PATH%\pretrained_models\strand_prior\strand echo [√] Neural Haircut Strand Prior验证成功 ) -IF NOT EXIST "%PIXIE_PATH%" echo 警告:缺少PIXIE模型 -IF NOT EXIST "%SAM_PATH%" echo 警告:缺少SAM模型 -IF NOT EXIST "%DINO_PATH%" echo 警告:缺少GroundingDINO模型 -IF NOT EXIST "%OPENPOSE_PATH%" echo 警告:缺少OpenPose模型 -IF NOT EXIST "%HYPERIQA_PATH%" echo 警告:缺少hyperIQA模型 +IF NOT EXIST "%PIXIE_PATH%" ( + echo [×] 缺少PIXIE模型 + SET /A VERIFY_FAILED+=1 +) ELSE ( + 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 ( echo.