diff --git a/download_resource.bat b/download_resource.bat index 2e7fadf..82c027f 100644 --- a/download_resource.bat +++ b/download_resource.bat @@ -13,6 +13,92 @@ SET "DINO_PATH=resource\Matte-Anything\groundingdino_swint_ogc.pth" SET "OPENPOSE_PATH=resource\openpose\models\pose\coco\pose_iter_584000.caffemodel" SET "HYPERIQA_PATH=resource\hyperIQA\pretrained\hyperIQA.pth" +REM 检查已有模型文件 +echo 正在检查已有模型文件... +SET MISSING_FILES=0 +SET /A TOTAL_FILES=7 +SET /A EXISTING_FILES=0 +SET MISSING_LIST= + +IF EXIST "%NEURAL_HAIRCUT_FOLDER_PATH%\pretrained_models\diffusion_prior\dif_ckpt.pt" ( + echo [√] Neural Haircut Diffusion Prior模型已存在 + SET /A EXISTING_FILES+=1 +) ELSE ( + echo [×] 缺少Neural Haircut Diffusion Prior模型 + SET /A MISSING_FILES+=1 + SET MISSING_LIST=!MISSING_LIST!Neural Haircut Diffusion Prior, +) + +IF EXIST "%NEURAL_HAIRCUT_FOLDER_PATH%\pretrained_models\strand_prior\strand_ckpt.pt" ( + echo [√] Neural Haircut Strand Prior模型已存在 + SET /A EXISTING_FILES+=1 +) ELSE ( + echo [×] 缺少Neural Haircut Strand Prior模型 + SET /A MISSING_FILES+=1 + SET MISSING_LIST=!MISSING_LIST!Neural Haircut Strand Prior, +) + +IF EXIST "%PIXIE_PATH%" ( + echo [√] PIXIE模型已存在 + SET /A EXISTING_FILES+=1 +) ELSE ( + echo [×] 缺少PIXIE模型 + SET /A MISSING_FILES+=1 + SET MISSING_LIST=!MISSING_LIST!PIXIE, +) + +IF EXIST "%SAM_PATH%" ( + echo [√] SAM模型已存在 + SET /A EXISTING_FILES+=1 +) ELSE ( + echo [×] 缺少SAM模型 + SET /A MISSING_FILES+=1 + SET MISSING_LIST=!MISSING_LIST!SAM, +) + +IF EXIST "%DINO_PATH%" ( + echo [√] GroundingDINO模型已存在 + SET /A EXISTING_FILES+=1 +) ELSE ( + echo [×] 缺少GroundingDINO模型 + SET /A MISSING_FILES+=1 + SET MISSING_LIST=!MISSING_LIST!GroundingDINO, +) + +IF EXIST "%MATTE_PATH%" ( + echo [√] Matte-Anything模型已存在 + SET /A EXISTING_FILES+=1 +) ELSE ( + echo [×] 缺少Matte-Anything模型 + SET /A MISSING_FILES+=1 + SET MISSING_LIST=!MISSING_LIST!Matte-Anything, +) + +IF EXIST "%OPENPOSE_PATH%" ( + echo [√] OpenPose模型已存在 + SET /A EXISTING_FILES+=1 +) ELSE ( + echo [×] 缺少OpenPose模型 + SET /A MISSING_FILES+=1 + SET MISSING_LIST=!MISSING_LIST!OpenPose, +) + +echo. +echo 已存在 %EXISTING_FILES%/%TOTAL_FILES% 个模型文件 +IF %MISSING_FILES% GTR 0 ( + echo 需要下载 %MISSING_FILES% 个模型文件 + echo 缺少的模型: !MISSING_LIST:~0,-2! + echo. + echo ************************************************************ + echo * 开始下载所需资源文件 * + echo * 下载时间取决于您的网络连接速度 * + echo * 请耐心等待... * + echo ************************************************************ +) ELSE ( + echo 所有模型文件已存在,无需下载 + exit /b 0 +) + echo ======================================= 基础环境配置 ============================================== echo. echo **************************************************************************************************