diff --git a/download_resource.bat b/download_resource.bat index 18861a4..ede28b0 100644 --- a/download_resource.bat +++ b/download_resource.bat @@ -1,47 +1,18 @@ @echo off setlocal enabledelayedexpansion -REM 错误处理 -:ERROR_HANDLER -IF %ERRORLEVEL% NEQ 0 ( - echo. - echo ************************************************************ - echo * 下载任务被中断 * - echo * 请检查网络连接后重新运行脚本 * - echo * 如果问题持续存在，请查看以上错误信息 * - echo ************************************************************ - exit /b 1 -) - -REM 设置当前目录 SET PROJECT_DIR=%~dp0 cd %PROJECT_DIR% -REM 创建resource目录结构 -echo 正在创建resource目录结构... -mkdir resource 2>nul -mkdir resource\NeuralHaircut 2>nul -mkdir resource\NeuralHaircut\PIXIE 2>nul -mkdir resource\Matte-Anything 2>nul -mkdir resource\openpose 2>nul -mkdir resource\hyperIQA 2>nul - -REM 验证已有文件 -echo 正在检查已有资源文件... -SET MISSING_FILES=0 -SET /A TOTAL_FILES=7 -SET /A EXISTING_FILES=0 -SET MISSING_LIST= - -REM 检查7-Zip是否安装 -where 7z >nul 2>nul -IF %ERRORLEVEL% NEQ 0 ( - echo 错误：未找到7-Zip - echo 请从 https://7-zip.org/ 下载并安装7-Zip - echo 并将其添加到系统环境变量PATH中 - exit /b 1 -) +echo ======================================= 基础环境配置 ============================================== +echo. +echo ************************************************************************************************** +echo * 基础环境配置 * +echo * 检查Python是否安装，更新pip，安装gdown * +echo * 请耐心等待... * +echo ************************************************************************************************** +echo. REM 检查Python是否安装 python --version >nul 2>nul IF %ERRORLEVEL% NEQ 0 ( @@ -75,6 +46,16 @@ set PYTHON_SCRIPTS_PATH=%PYTHON_PATH:python.exe=Scripts% set PYTHON_USER_SCRIPTS=%USERPROFILE%\AppData\Roaming\Python\Python3*\Scripts set PATH=%PYTHON_SCRIPTS_PATH%;%PYTHON_USER_SCRIPTS%;%PATH% +REM 检查7-Zip是否安装 +where 7z >nul 2>nul +IF %ERRORLEVEL% NEQ 0 ( + echo 错误：未找到7-Zip + echo 请从 https://7-zip.org/ 下载并安装7-Zip + echo 并将其添加到系统环境变量PATH中 + exit /b 1 +) + + REM 验证gdown是否可用 echo 正在验证gdown安装... python -c "import gdown" >nul 2>nul @@ -86,50 +67,57 @@ IF %ERRORLEVEL% NEQ 0 ( echo gdown安装成功！ echo. -REM Download Neural Haircut files +echo ======================================= 开始下载所需资源文件 ======================================= echo. -echo ************************************************************ -echo * 开始下载所需资源文件 * -echo * 下载时间取决于您的网络连接速度 * -echo * 请耐心等待... * -echo ************************************************************ +echo ************************************************************************************************** +echo * 开始下载所需资源文件 * +echo * 下载时间取决于您的网络连接速度 * +echo * 请耐心等待... * +echo ************************************************************************************************** echo. -echo [1/7] 正在下载Neural Haircut文件... +echo [1/8] 正在下载Neural Haircut文件... +mkdir %PROJECT_DIR%/resource/NeuralHaircut cd %PROJECT_DIR%/resource/NeuralHaircut python -m gdown --folder "https://drive.google.com/drive/folders/1TCdJ0CKR3Q6LviovndOkJaKm8S1T9F_8" +cd %PROJECT_DIR% -REM 更新 diffusion_prior 模型 -cd %PROJECT_DIR%\resource\NeuralHaircut\pretrained_models\diffusion_prior -echo 正在下载Neural Haircut扩散先验模型文件... +echo [2/8] 正在下载Neural Haircut扩散先验模型文件... +mkdir %PROJECT_DIR%/resource/NeuralHaircut/pretrained_models/diffusion_prior +cd %PROJECT_DIR%/resource/NeuralHaircut/pretrained_models/diffusion_prior python -m gdown "1_9EOUXHayKiGH5nkrayncln3d6m1uV7f" +cd %PROJECT_DIR% -REM 下载 PIXIE 文件 +echo [3/8] 正在下载Neural HaircutPIXIE模型文件... +mkdir %PROJECT_DIR%/resource/NeuralHaircut/PIXIE cd %PROJECT_DIR%/resource/NeuralHaircut/PIXIE -echo 正在下载Neural HaircutPIXIE模型文件... python -m gdown "1mPcGu62YPc4MdkT8FFiOCP629xsENHZf" && tar -xvzf pixie_data.tar.gz ./ && rm pixie_data.tar.gz +cd %PROJECT_DIR% -REM 下载 hyperIQA 模型 +echo [4/8] 正在下载hyperIQA模型文件... +mkdir %PROJECT_DIR%/resource/hyperIQA/pretrained cd %PROJECT_DIR%/resource/hyperIQA/pretrained -echo 正在下载hyperIQA模型文件... python -m gdown "1OOUmnbvpGea0LIGpIWEbOyxfWx6UCiiE" cd %PROJECT_DIR% -REM Download Matte-Anything files +echo [5/8] 正在下载Matte-Anything扩散先验模型文件... +mkdir %PROJECT_DIR%/resource/Matte-Anything/pretrained cd %PROJECT_DIR%/resource/Matte-Anything/pretrained -echo 正在下载Matte-Anything扩散先验模型文件... -wget https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth -echo 正在下载Matte-AnythingGroundingDINO模型文件... -wget https://github.com/IDEA-Research/GroundingDINO/releases/download/v0.1.0-alpha/groundingdino_swint_ogc.pth -echo 正在下载Matte-Anything模型文件... +wget "https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth" +echo [6/8] 正在下载Matte-AnythingGroundingDINO模型文件... +mkdir %PROJECT_DIR%/resource/Matte-Anything/GroundingDINO +cd %PROJECT_DIR%/resource/Matte-Anything/GroundingDINO +wget "https://github.com/IDEA-Research/GroundingDINO/releases/download/v0.1.0-alpha/groundingdino_swint_ogc.pth" +echo [7/8] 正在下载Matte-Anything模型文件... python -m gdown "1d97oKuITCeWgai2Tf3iNilt6rMSSYzkW" - -REM Download OpenPose files -cd %PROJECT_DIR%/resource/openpose -echo 正在下载OpenPose模型文件... -python -m gdown "1Yn03cKKfVOq4qXmgBMQD20UMRRRkd_tV" && tar -xvzf models.tar.gz && rm models.tar.gz - cd %PROJECT_DIR% + +echo [8/8] 正在下载OpenPose模型文件... +mkdir %PROJECT_DIR%/resource/openpose/models/pose/coco +cd %PROJECT_DIR%/resource/openpose/models/pose/coco +python -m gdown "1Yn03cKKfVOq4qXmgBMQD20UMRRRkd_tV" && tar -xvzf models.tar.gz && rm models.tar.gz +cd %PROJECT_DIR% + echo 所有模型文件下载完成！ diff --git a/install.bat b/install.bat index e8bcee3..cd0e7a9 100644 --- a/install.bat +++ b/install.bat @@ -165,72 +165,14 @@ IF NOT EXIST "%PROJECT_DIR%\resource" ( echo resource/ echo ├── NeuralHaircut/ echo │ ├── diffusion_prior/ - echo │ │ └── model.pt + echo │ │ └── dif_ckpt.pt echo │ ├── strand_prior/ - echo │ │ └── model.pt + echo │ │ └── strand_ckpt.pt echo │ └── PIXIE/ echo │ └── pixie_data/ echo ├── Matte-Anything/ - echo │ ├── sam_vit_h_4b8939.pth - echo │ ├── groundingdino_swint_ogc.pth - echo │ └── model.pth echo ├── openpose/ - echo │ └── models/ - echo │ └── pose/ - echo │ └── coco/ - echo │ └── pose_iter_584000.caffemodel echo └── hyperIQA/ - echo └── pretrained/ - echo └── hyperIQA.pth - exit /b 1 -) - -REM 检查模型文件是否存在 -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\strand_prior\model.pt" ( - echo ERROR: Neural Haircut Strand Prior model not found - echo Please run download_resource.bat first - exit /b 1 -) - -IF NOT EXIST "%PROJECT_DIR%\resource\NeuralHaircut\PIXIE\pixie_data\pixie_data" ( - echo ERROR: PIXIE model not found - echo Please run download_resource.bat first - exit /b 1 -) - -IF NOT EXIST "%PROJECT_DIR%\resource\Matte-Anything\sam_vit_h_4b8939.pth" ( - echo ERROR: SAM model not found - echo Please run download_resource.bat first - exit /b 1 -) - -IF NOT EXIST "%PROJECT_DIR%\resource\Matte-Anything\groundingdino_swint_ogc.pth" ( - echo ERROR: GroundingDINO model not found - echo Please run download_resource.bat first - exit /b 1 -) - -IF NOT EXIST "%PROJECT_DIR%\resource\Matte-Anything\model.pth" ( - echo ERROR: Matte-Anything model not found - echo Please run download_resource.bat first - exit /b 1 -) - -IF NOT EXIST "%PROJECT_DIR%\resource\openpose\models\pose\coco\pose_iter_584000.caffemodel" ( - echo ERROR: OpenPose model not found - echo Please run download_resource.bat first - exit /b 1 -) - -IF NOT EXIST "%PROJECT_DIR%\resource\hyperIQA\pretrained\hyperIQA.pth" ( - echo ERROR: hyperIQA model not found - echo Please run download_resource.bat first exit /b 1 ) @@ -247,23 +189,11 @@ 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 /E /I /Y "%PROJECT_DIR%\resource\NeuralHaircut\*" "%PROJECT_DIR%\ext\NeuralHaircut\" - -REM PIXIE -cd "%PROJECT_DIR%\ext\PIXIE" -xcopy /E /I /Y "%PROJECT_DIR%\resource\NeuralHaircut\PIXIE\*" . - -REM Matte-Anything -xcopy /Y "%PROJECT_DIR%\resource\Matte-Anything\sam_vit_h_4b8939.pth" "%PROJECT_DIR%\ext\Matte-Anything\pretrained\" -xcopy /Y "%PROJECT_DIR%\resource\Matte-Anything\groundingdino_swint_ogc.pth" "%PROJECT_DIR%\ext\Matte-Anything\pretrained\" -xcopy /Y "%PROJECT_DIR%\resource\Matte-Anything\model.pth" "%PROJECT_DIR%\ext\Matte-Anything\" - -REM OpenPose -xcopy /Y "%PROJECT_DIR%\resource\openpose\models\pose_iter_584000.caffemodel" "%PROJECT_DIR%\ext\openpose\models\pose\coco\" - -REM hyperIQA -xcopy /Y "%PROJECT_DIR%\resource\hyperIQA\pretrained\hyperIQA.pth" "%PROJECT_DIR%\ext\hyperIQA\pretrained\" +xcopy /E /I /Y "%PROJECT_DIR%\resource\NeuralHaircut\PIXIE\*" "%PROJECT_DIR%\ext\PIXIE" +xcopy /Y "%PROJECT_DIR%\resource\Matte-Anything\*" "%PROJECT_DIR%\ext\Matte-Anything\pretrained\" +xcopy /Y "%PROJECT_DIR%\resource\openpose\*" "%PROJECT_DIR%\ext\openpose\*" +xcopy /Y "%PROJECT_DIR%\resource\hyperIQA\*" "%PROJECT_DIR%\ext\hyperIQA\*" REM 安装 PIXIE 环境 CALL activate_pixie-env.bat