diff --git a/download_resource.bat b/download_resource.bat index 7f3d8ff..b716fa6 100644 --- a/download_resource.bat +++ b/download_resource.bat @@ -72,59 +72,79 @@ echo 正在检查已有资源文件... SET MISSING_FILES=0 SET /A TOTAL_FILES=6 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模型已存在 SET /A EXISTING_FILES+=1 ) ELSE ( echo [×] 缺少Neural Haircut模型 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模型数据已存在 SET /A EXISTING_FILES+=1 ) ELSE ( echo [×] 缺少PIXIE模型数据 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模型已存在 SET /A EXISTING_FILES+=1 ) ELSE ( echo [×] 缺少SAM模型 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模型已存在 SET /A EXISTING_FILES+=1 ) ELSE ( echo [×] 缺少GroundingDINO模型 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模型已存在 SET /A EXISTING_FILES+=1 ) ELSE ( echo [×] 缺少OpenPose模型 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模型已存在 SET /A EXISTING_FILES+=1 ) ELSE ( echo [×] 缺少hyperIQA模型 SET /A MISSING_FILES+=1 + SET MISSING_LIST=!MISSING_LIST!hyperIQA模型, ) echo. echo 已存在 %EXISTING_FILES%/%TOTAL_FILES% 个模型文件 IF %MISSING_FILES% GTR 0 ( echo 需要下载 %MISSING_FILES% 个模型文件 + echo 缺少的模型: !MISSING_LIST:~0,-2! echo. echo ************************************************************ echo * 开始下载所需资源文件 * @@ -214,7 +234,7 @@ IF EXIST "model.pth" ( REM 下载 OpenPose 模型 cd ..\openpose echo 正在下载OpenPose模型... -IF EXIST "models\pose_iter_584000.caffemodel" ( +IF EXIST "models\pose\coco\pose_iter_584000.caffemodel" ( echo OpenPose模型已存在,跳过下载... ) ELSE ( echo 正在下载OpenPose模型...