diff --git a/download_resource.bat b/download_resource.bat index ebce497..4d255b6 100644 --- a/download_resource.bat +++ b/download_resource.bat @@ -24,7 +24,7 @@ SET "HYPERIQA_MODEL_URL=1OOUmnbvpGea0LIGpIWEbOyxfWx6UCiiE" REM 定义文件路径 SET "NEURAL_HAIRCUT_FOLDER_PATH=resource\NeuralHaircut" -SET "PIXIE_PATH=resource\NeuralHaircut\PIXIE\pixie_data.tar.gz" +SET "PIXIE_PATH=resource\NeuralHaircut\PIXIE\pixie_data\pixie_data" SET "MATTE_PATH=resource\Matte-Anything\model.pth" SET "SAM_PATH=resource\Matte-Anything\sam_vit_h_4b8939.pth" SET "DINO_PATH=resource\Matte-Anything\groundingdino_swint_ogc.pth" @@ -98,7 +98,6 @@ REM 创建resource目录结构 mkdir resource 2>nul cd resource mkdir NeuralHaircut 2>nul -mkdir NeuralHaircut\pretrained_models 2>nul mkdir Matte-Anything 2>nul mkdir openpose 2>nul mkdir hyperIQA 2>nul @@ -106,12 +105,12 @@ mkdir hyperIQA 2>nul REM 验证已有文件 echo 正在检查已有资源文件... SET MISSING_FILES=0 -SET /A TOTAL_FILES=8 +SET /A TOTAL_FILES=7 SET /A EXISTING_FILES=0 SET MISSING_LIST= REM Neural Haircut Diffusion Prior 检查 -IF EXIST "%NEURAL_HAIRCUT_FOLDER_PATH%\diffusion_prior\dif_ckpt.pt" ( +IF EXIST "%NEURAL_HAIRCUT_FOLDER_PATH%\pretrained_models\diffusion_prior\dif_ckpt.pt" ( echo [√] Neural Haircut Diffusion Prior已存在 SET /A EXISTING_FILES+=1 ) ELSE ( @@ -121,7 +120,7 @@ IF EXIST "%NEURAL_HAIRCUT_FOLDER_PATH%\diffusion_prior\dif_ckpt.pt" ( ) REM Neural Haircut Strand Prior 检查 -IF EXIST "%NEURAL_HAIRCUT_FOLDER_PATH%\strand_prior\strand_ckpt.pt" ( +IF EXIST "%NEURAL_HAIRCUT_FOLDER_PATH%\pretrained_models\strand_prior\strand_ckpt.pt" ( echo [√] Neural Haircut Strand Prior已存在 SET /A EXISTING_FILES+=1 ) ELSE ( @@ -208,6 +207,7 @@ echo 正在下载Neural Haircut预训练文件夹... python -m gdown --folder "%NEURAL_HAIRCUT_FOLDER_URL%" REM 更新 diffusion_prior 模型 +mkdir pretrained_models\diffusion_prior 2>nul cd pretrained_models\diffusion_prior echo 正在更新Neural Haircut Diffusion Prior模型... python -m gdown "%NEURAL_HAIRCUT_MODEL_URL%" @@ -263,14 +263,14 @@ echo 正在验证下载文件... cd %PROJECT_DIR% SET VERIFY_FAILED=0 -IF NOT EXIST "%NEURAL_HAIRCUT_FOLDER_PATH%\diffusion_prior\dif_ckpt.pt" ( +IF NOT EXIST "%NEURAL_HAIRCUT_FOLDER_PATH%\pretrained_models\diffusion_prior\dif_ckpt.pt" ( echo [×] 缺少Neural Haircut Diffusion Prior SET /A VERIFY_FAILED+=1 ) ELSE ( echo [√] Neural Haircut Diffusion Prior验证成功 ) -IF NOT EXIST "%NEURAL_HAIRCUT_FOLDER_PATH%\strand_prior\strand_ckpt.pt" ( +IF NOT EXIST "%NEURAL_HAIRCUT_FOLDER_PATH%\pretrained_models\strand_prior\strand_ckpt.pt" ( echo [×] 缺少Neural Haircut Strand Prior SET /A VERIFY_FAILED+=1 ) ELSE ( diff --git a/install.bat b/install.bat index 15c0d59..e8bcee3 100644 --- a/install.bat +++ b/install.bat @@ -164,11 +164,10 @@ IF NOT EXIST "%PROJECT_DIR%\resource" ( echo. echo resource/ echo ├── NeuralHaircut/ - echo │ ├── pretrained_models/ - echo │ │ ├── diffusion_prior/ - echo │ │ │ └── dif_ckpt.pt - echo │ │ └── strand_prior/ - echo │ │ └── strand_ckpt.pt + echo │ ├── diffusion_prior/ + echo │ │ └── model.pt + echo │ ├── strand_prior/ + echo │ │ └── model.pt echo │ └── PIXIE/ echo │ └── pixie_data/ echo ├── Matte-Anything/ @@ -187,13 +186,13 @@ IF NOT EXIST "%PROJECT_DIR%\resource" ( ) REM 检查模型文件是否存在 -IF NOT EXIST "%PROJECT_DIR%\resource\NeuralHaircut\pretrained_models\diffusion_prior\dif_ckpt.pt" ( +IF NOT EXIST "%PROJECT_DIR%\resource\NeuralHaircut\diffusion_prior\model.pt" ( echo ERROR: Neural Haircut Diffusion Prior model not found echo Please run download_resource.bat first exit /b 1 ) -IF NOT EXIST "%PROJECT_DIR%\resource\NeuralHaircut\pretrained_models\strand_prior\strand_ckpt.pt" ( +IF NOT EXIST "%PROJECT_DIR%\resource\NeuralHaircut\strand_prior\model.pt" ( echo ERROR: Neural Haircut Strand Prior model not found echo Please run download_resource.bat first exit /b 1