diff --git a/download_resource.bat b/download_resource.bat index d5152a7..6c87751 100644 --- a/download_resource.bat +++ b/download_resource.bat @@ -15,11 +15,11 @@ IF EXIST "hyperIQA" rmdir /s /q "hyperIQA" REM 定义模型信息 SET "MODEL_NEURAL_HAIRCUT_NAME=Neural Haircut模型" -SET "MODEL_NEURAL_HAIRCUT_PATH=resource\NeuralHaircut\diffusion_prior\model.pt" +SET "MODEL_NEURAL_HAIRCUT_PATH=resource\NeuralHaircut\pretrained_models\diffusion_prior\model.pt" SET "MODEL_NEURAL_HAIRCUT_URL=1_9EOUXHayKiGH5nkrayncln3d6m1uV7f" SET "MODEL_PIXIE_NAME=PIXIE模型数据" -SET "MODEL_PIXIE_PATH=resource\NeuralHaircut\PIXIE\pixie_data" +SET "MODEL_PIXIE_PATH=resource\PIXIE\pixie_data.tar.gz" SET "MODEL_PIXIE_URL=1mPcGu62YPc4MdkT8FFiOCP629xsENHZf" SET "MODEL_MATTE_NAME=Matte-Anything模型" @@ -108,10 +108,10 @@ echo. REM 创建resource目录结构 mkdir resource 2>nul cd resource -mkdir NeuralHaircut\diffusion_prior 2>nul -mkdir NeuralHaircut\PIXIE 2>nul +mkdir NeuralHaircut\pretrained_models\diffusion_prior 2>nul +mkdir PIXIE 2>nul mkdir Matte-Anything 2>nul -mkdir openpose\models\pose\coco 2>nul +mkdir openpose\models 2>nul mkdir hyperIQA\pretrained 2>nul REM 验证已有文件 @@ -259,7 +259,7 @@ IF EXIST "%MODEL_MATTE_PATH%" ( ) REM 下载 OpenPose 模型 -cd ..\openpose\models\pose\coco +cd ..\openpose\models echo 正在下载OpenPose模型... IF EXIST "%MODEL_OPENPOSE_PATH%" ( echo %MODEL_OPENPOSE_NAME%已存在,跳过下载... @@ -316,7 +316,7 @@ IF NOT EXIST "%MODEL_NEURAL_HAIRCUT_PATH%" ( IF NOT EXIST "%MODEL_PIXIE_PATH%" echo 警告:缺少%MODEL_PIXIE_NAME% IF NOT EXIST "%MODEL_SAM_PATH%" echo 警告:缺少%MODEL_SAM_NAME% IF NOT EXIST "%MODEL_DINO_PATH%" echo 警告:缺少%MODEL_DINO_NAME% -IF NOT EXIST "openpose\models\pose\coco" echo 警告:缺少OpenPose姿态模型 +IF NOT EXIST "%MODEL_OPENPOSE_PATH%" echo 警告:缺少%MODEL_OPENPOSE_NAME% IF NOT EXIST "%MODEL_HYPERIQA_PATH%" echo 警告:缺少%MODEL_HYPERIQA_NAME% IF %VERIFY_FAILED% GTR 0 ( diff --git a/install.bat b/install.bat index 805ef18..47ccfbf 100644 --- a/install.bat +++ b/install.bat @@ -167,11 +167,22 @@ IF NOT EXIST "%PROJECT_DIR%\resource" ( REM 复制模型文件到对应位置 echo 正在复制模型文件... + +REM 创建必要的目录 +mkdir "%PROJECT_DIR%\ext\NeuralHaircut\pretrained_models\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\diffusion_prior\model.pt" "%PROJECT_DIR%\ext\NeuralHaircut\pretrained_models\diffusion_prior\" +xcopy /Y "%PROJECT_DIR%\resource\NeuralHaircut\pretrained_models\diffusion_prior\model.pt" "%PROJECT_DIR%\ext\NeuralHaircut\pretrained_models\diffusion_prior\" REM PIXIE -xcopy /E /I /Y "%PROJECT_DIR%\resource\NeuralHaircut\PIXIE\pixie_data" "%PROJECT_DIR%\ext\PIXIE\" +cd "%PROJECT_DIR%\ext\PIXIE" +7z x "%PROJECT_DIR%\resource\PIXIE\pixie_data.tar.gz" -y +7z x pixie_data.tar -y +del pixie_data.tar REM Matte-Anything xcopy /Y "%PROJECT_DIR%\resource\Matte-Anything\sam_vit_h_4b8939.pth" "%PROJECT_DIR%\ext\Matte-Anything\pretrained\" @@ -191,9 +202,6 @@ echo 正在安装PIXIE依赖... pip install pyyaml==5.4.1 pip install git+https://github.com/1adrianb/face-alignment.git@54623537fd9618ca7c15688fd85aba706ad92b59 -REM 复制PIXIE数据 -xcopy /E /I /Y "%PROJECT_DIR%\ext\PIXIE\*" "%PROJECT_DIR%\ext\NeuralHaircut\PIXIE\" - REM 安装 Matte-Anything 环境 CALL activate_matte_anything.bat cd %PROJECT_DIR%\ext\Matte-Anything