diff --git a/download_resource.bat b/download_resource.bat index 93c5812..b45eced 100644 --- a/download_resource.bat +++ b/download_resource.bat @@ -39,7 +39,7 @@ SET "MODEL_DINO_PATH=resource\Matte-Anything\groundingdino_swint_ogc.pth" SET "MODEL_DINO_URL=https://github.com/IDEA-Research/GroundingDINO/releases/download/v0.1.0-alpha/groundingdino_swint_ogc.pth" SET "MODEL_OPENPOSE_NAME=OpenPose模型" -SET "MODEL_OPENPOSE_PATH=resource\openpose\models\pose_iter_584000.caffemodel" +SET "MODEL_OPENPOSE_PATH=resource\openpose\models\pose\coco\pose_iter_584000.caffemodel" SET "MODEL_OPENPOSE_URL=1Yn03cKKfVOq4qXmgBMQD20UMRRRkd_tV" SET "MODEL_HYPERIQA_NAME=hyperIQA模型" @@ -115,7 +115,7 @@ cd resource mkdir NeuralHaircut\pretrained_models\diffusion_prior 2>nul mkdir PIXIE 2>nul mkdir Matte-Anything 2>nul -mkdir openpose\models 2>nul +mkdir openpose\models\pose\coco 2>nul mkdir hyperIQA\pretrained 2>nul REM 验证已有文件 @@ -257,7 +257,7 @@ IF EXIST "%MODEL_MATTE_PATH%" ( ) REM 下载 OpenPose 模型 -cd ..\openpose\models +cd ..\openpose\models\pose\coco echo 正在下载OpenPose模型... IF EXIST "%MODEL_OPENPOSE_PATH%" ( echo %MODEL_OPENPOSE_NAME%已存在,跳过下载... diff --git a/install.bat b/install.bat index c7ea525..dd25f81 100644 --- a/install.bat +++ b/install.bat @@ -127,14 +127,10 @@ git clone https://github.com/SSL92/hyperIQA cd .. REM 创建环境 -CALL "%MICROMAMBA_EXE%" create -p %MAMBA_ROOT_PREFIX%\envs\gaussian_splatting_hair python=3.8 pytorch=2.0.0 torchvision pytorch-cuda=11.8 cmake ninja setuptools=58.2.0 -c pytorch -c nvidia -c conda-forge -y -CALL "%MICROMAMBA_EXE%" create -p %MAMBA_ROOT_PREFIX%\envs\matte_anything pytorch=2.0.0 pytorch-cuda=11.8 torchvision tensorboard timm=0.5.4 opencv=4.5.3 mkl=2024.0 setuptools=58.2.0 easydict wget scikit-image gradio=3.46.1 fairscale supervision==0.22.0 -c pytorch -c nvidia -c conda-forge -y -CALL "%MICROMAMBA_EXE%" create -p %MAMBA_ROOT_PREFIX%\envs\openpose python=3.8 cmake=3.20 -c conda-forge -y -CALL "%MICROMAMBA_EXE%" create -p %MAMBA_ROOT_PREFIX%\envs\pixie-env python=3.8 pytorch=2.0.0 torchvision==0.15.0 torchaudio==2.0.0 pytorch-cuda=11.8 fvcore pytorch3d==0.7.5 kornia matplotlib -c pytorch -c nvidia -c fvcore -c conda-forge -c pytorch3d -c bottler -c iopath -y +CALL "%MICROMAMBA_EXE%" env create -f environment.yml REM 安装 gaussian_splatting_hair 环境 CALL "%MICROMAMBA_EXE%" activate -p %MAMBA_ROOT_PREFIX%\envs\gaussian_splatting_hair -pip install -r requirements.txt REM 安装外部库 cd %PROJECT_DIR%\ext\pytorch3d @@ -156,14 +152,23 @@ IF NOT EXIST "%PROJECT_DIR%\resource" ( echo. echo resource/ echo ├── NeuralHaircut/ - echo │ ├── diffusion_prior/ - echo │ └── PIXIE/ + echo │ ├── pretrained_models/ + 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 echo │ └── model.pth - echo └── openpose/ - echo └── models/ + echo ├── openpose/ + echo │ └── models/ + echo │ └── pose/ + echo │ └── coco/ + echo │ └── pose_iter_584000.caffemodel + echo └── hyperIQA/ + echo └── pretrained/ + echo └── hyperIQA.pth exit /b 1 )