diff --git a/download_resource.bat b/download_resource.bat index b45eced..706bbf1 100644 --- a/download_resource.bat +++ b/download_resource.bat @@ -15,15 +15,15 @@ IF EXIST "hyperIQA" rmdir /s /q "hyperIQA" REM 定义模型信息 SET "MODEL_NEURAL_HAIRCUT_NAME=Neural Haircut模型" -SET "MODEL_NEURAL_HAIRCUT_PATH=resource\NeuralHaircut\pretrained_models\diffusion_prior\model.pt" +SET "MODEL_NEURAL_HAIRCUT_PATH=resource\NeuralHaircut\diffusion_prior\model.pt" SET "MODEL_NEURAL_HAIRCUT_URL=1_9EOUXHayKiGH5nkrayncln3d6m1uV7f" SET "MODEL_PIXIE_NAME=PIXIE模型数据" -SET "MODEL_PIXIE_PATH=resource\PIXIE\pixie_data" +SET "MODEL_PIXIE_PATH=resource\NeuralHaircut\PIXIE\pixie_data" SET "MODEL_PIXIE_URL=1mPcGu62YPc4MdkT8FFiOCP629xsENHZf" SET "MODEL_PIXIE_FACE_NAME=PIXIE Face模型" -SET "MODEL_PIXIE_FACE_PATH=resource\PIXIE\pixie_data\pixie_data" +SET "MODEL_PIXIE_FACE_PATH=resource\NeuralHaircut\PIXIE\pixie_data\pixie_data" SET "MODEL_PIXIE_FACE_URL=https://github.com/yfeng95/PIXIE/raw/master/fetch_model.sh" SET "MODEL_MATTE_NAME=Matte-Anything模型" @@ -112,8 +112,8 @@ echo. REM 创建resource目录结构 mkdir resource 2>nul cd resource -mkdir NeuralHaircut\pretrained_models\diffusion_prior 2>nul -mkdir PIXIE 2>nul +mkdir NeuralHaircut\diffusion_prior 2>nul +mkdir NeuralHaircut\PIXIE 2>nul mkdir Matte-Anything 2>nul mkdir openpose\models\pose\coco 2>nul mkdir hyperIQA\pretrained 2>nul @@ -219,7 +219,7 @@ IF EXIST "%MODEL_NEURAL_HAIRCUT_PATH%" ( ) REM 下载 PIXIE 文件 -cd PIXIE +cd NeuralHaircut\PIXIE IF EXIST "pixie_data.tar" ( echo PIXIE模型已存在,跳过下载... ) ELSE ( diff --git a/install.bat b/install.bat index dd25f81..110d55d 100644 --- a/install.bat +++ b/install.bat @@ -152,11 +152,10 @@ IF NOT EXIST "%PROJECT_DIR%\resource" ( echo. echo resource/ echo ├── NeuralHaircut/ - echo │ ├── pretrained_models/ - echo │ │ └── diffusion_prior/ - echo │ │ └── model.pt - echo ├── PIXIE/ - echo │ └── pixie_data/ + echo │ ├── diffusion_prior/ + echo │ │ └── model.pt + echo │ └── PIXIE/ + echo │ └── pixie_data/ echo ├── Matte-Anything/ echo │ ├── sam_vit_h_4b8939.pth echo │ ├── groundingdino_swint_ogc.pth @@ -176,18 +175,18 @@ REM 复制模型文件到对应位置 echo 正在复制模型文件... REM 创建必要的目录 -mkdir "%PROJECT_DIR%\ext\NeuralHaircut\pretrained_models\diffusion_prior" 2>nul +mkdir "%PROJECT_DIR%\ext\NeuralHaircut\diffusion_prior" 2>nul mkdir "%PROJECT_DIR%\ext\PIXIE" 2>nul mkdir "%PROJECT_DIR%\ext\Matte-Anything\pretrained" 2>nul mkdir "%PROJECT_DIR%\ext\openpose\models\pose\coco" 2>nul mkdir "%PROJECT_DIR%\ext\hyperIQA\pretrained" 2>nul REM Neural Haircut -xcopy /Y "%PROJECT_DIR%\resource\NeuralHaircut\pretrained_models\diffusion_prior\model.pt" "%PROJECT_DIR%\ext\NeuralHaircut\pretrained_models\diffusion_prior\" +xcopy /Y "%PROJECT_DIR%\resource\NeuralHaircut\diffusion_prior\model.pt" "%PROJECT_DIR%\ext\NeuralHaircut\diffusion_prior\" REM PIXIE cd "%PROJECT_DIR%\ext\PIXIE" -7z x "%PROJECT_DIR%\resource\PIXIE\pixie_data.tar.gz" -y +7z x "%PROJECT_DIR%\resource\NeuralHaircut\PIXIE\pixie_data.tar.gz" -y 7z x pixie_data.tar -y del pixie_data.tar