diff --git a/download_resource.bat b/download_resource.bat index 4d255b6..5a8a02f 100644 --- a/download_resource.bat +++ b/download_resource.bat @@ -204,16 +204,24 @@ IF DEFINED HTTP_PROXY ( REM 下载 Neural Haircut 文件 cd resource\NeuralHaircut echo 正在下载Neural Haircut预训练文件夹... -python -m gdown --folder "%NEURAL_HAIRCUT_FOLDER_URL%" - -REM 更新 diffusion_prior 模型 +mkdir pretrained_models 2>nul mkdir pretrained_models\diffusion_prior 2>nul +mkdir pretrained_models\strand_prior 2>nul + +REM 下载 diffusion_prior 模型 cd pretrained_models\diffusion_prior -echo 正在更新Neural Haircut Diffusion Prior模型... -python -m gdown "%NEURAL_HAIRCUT_MODEL_URL%" +echo 正在下载Neural Haircut Diffusion Prior模型... +python -m gdown "1OBR0Vzb_w5SNc1jmoScA6L8SqVniJGOr" -O dif_ckpt.pt +cd .. + +REM 下载 strand_prior 模型 +cd strand_prior +echo 正在下载Neural Haircut Strand Prior模型... +python -m gdown "1DESwUb-nsmi38VCDvnBwpd9kjcWONNT6" -O strand_ckpt.pt cd ..\.. REM 下载 PIXIE 文件 +mkdir PIXIE 2>nul cd PIXIE IF EXIST "pixie_data.tar" ( echo PIXIE模型已存在,跳过下载...