diff --git a/download_resource.bat b/download_resource.bat index 5ace385..3cf7503 100644 --- a/download_resource.bat +++ b/download_resource.bat @@ -15,12 +15,6 @@ python -m pip install --upgrade pip REM 安装gdown python -m pip install gdown -REM 安装7z -python -m pip install py7zr - -REM 安装tar -python -m pip install tar - mkdir %PROJECT_DIR%\resource\NeuralHaircut\ cd %PROJECT_DIR%\resource\NeuralHaircut\ python -m gdown --folder https://drive.google.com/drive/folders/1TCdJ0CKR3Q6LviovndOkJaKm8S1T9F_8 diff --git a/install.bat b/install.bat index 5b19c39..78a78d3 100644 --- a/install.bat +++ b/install.bat @@ -186,27 +186,47 @@ REM 复制模型文件到对应位置 echo 正在复制模型文件... REM 创建必要的目录 -mkdir "%PROJECT_DIR%\ext\NeuralHaircut" 2>nul -mkdir "%PROJECT_DIR%\ext\NeuralHaircut\pretrained_models" 2>nul mkdir "%PROJECT_DIR%\ext\NeuralHaircut\pretrained_models\diffusion_prior" 2>nul mkdir "%PROJECT_DIR%\ext\NeuralHaircut\pretrained_models\strand_prior" 2>nul -mkdir "%PROJECT_DIR%\ext\PIXIE" 2>nul +mkdir "%PROJECT_DIR%\ext\NeuralHaircut\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\openpose\models" 2>nul mkdir "%PROJECT_DIR%\ext\hyperIQA\pretrained" 2>nul -REM 复制NeuralHaircut相关文件 -xcopy /Y "%PROJECT_DIR%\resource\NeuralHaircut\*" "%PROJECT_DIR%\ext\NeuralHaircut\" -xcopy /E /I /Y "%PROJECT_DIR%\resource\NeuralHaircut\PIXIE\*" "%PROJECT_DIR%\ext\PIXIE" +REM 复制 NeuralHaircut 相关文件 +xcopy /Y "%PROJECT_DIR%\resource\NeuralHaircut\pretrained_models\diffusion_prior\dif_ckpt.pt" ^ + "%PROJECT_DIR%\ext\NeuralHaircut\pretrained_models\diffusion_prior\" +xcopy /Y "%PROJECT_DIR%\resource\NeuralHaircut\pretrained_models\strand_prior\strand_ckpt.pt" ^ + "%PROJECT_DIR%\ext\NeuralHaircut\pretrained_models\strand_prior\" +xcopy /Y "%PROJECT_DIR%\resource\NeuralHaircut\PIXIE\pixie_data.tar.gz" ^ + "%PROJECT_DIR%\ext\NeuralHaircut\PIXIE\" -REM 复制Matte-Anything相关文件 -xcopy /Y "%PROJECT_DIR%\resource\Matte-Anything\*" "%PROJECT_DIR%\ext\Matte-Anything\" +REM 复制 Matte-Anything 相关文件 +xcopy /Y "%PROJECT_DIR%\resource\Matte-Anything\pretrained\ViTMatte_B_DIS.pth" ^ + "%PROJECT_DIR%\ext\Matte-Anything\pretrained\" -REM 复制OpenPose相关文件 -xcopy /E /I /Y "%PROJECT_DIR%\resource\openpose\*" "%PROJECT_DIR%\ext\openpose\" +REM 复制 OpenPose 相关文件 +xcopy /Y "%PROJECT_DIR%\resource\openpose\models\models.tar.gz" ^ + "%PROJECT_DIR%\ext\openpose\models\" -REM 复制hyperIQA相关文件 -xcopy /Y "%PROJECT_DIR%\resource\hyperIQA\*" "%PROJECT_DIR%\ext\hyperIQA\" +REM 复制 hyperIQA 相关文件 +xcopy /Y "%PROJECT_DIR%\resource\hyperIQA\pretrained\koniq_pretrained.pkl" ^ + "%PROJECT_DIR%\ext\hyperIQA\pretrained\" + +REM 解压缩文件 +cd "%PROJECT_DIR%\ext\NeuralHaircut\PIXIE" +if exist "pixie_data.tar.gz" ( + tar -xvzf pixie_data.tar.gz + del pixie_data.tar.gz +) + +cd "%PROJECT_DIR%\ext\openpose\models" +if exist "models.tar.gz" ( + tar -xvzf models.tar.gz + del models.tar.gz +) + +cd "%PROJECT_DIR%" REM 安装 PIXIE 环境 CALL activate_pixie-env.bat