Update download_resource.bat
This commit is contained in:
parent
4376af31d3
commit
b9348b6033
@ -204,16 +204,24 @@ IF DEFINED HTTP_PROXY (
|
|||||||
REM 下载 Neural Haircut 文件
|
REM 下载 Neural Haircut 文件
|
||||||
cd resource\NeuralHaircut
|
cd resource\NeuralHaircut
|
||||||
echo 正在下载Neural Haircut预训练文件夹...
|
echo 正在下载Neural Haircut预训练文件夹...
|
||||||
python -m gdown --folder "%NEURAL_HAIRCUT_FOLDER_URL%"
|
mkdir pretrained_models 2>nul
|
||||||
|
|
||||||
REM 更新 diffusion_prior 模型
|
|
||||||
mkdir pretrained_models\diffusion_prior 2>nul
|
mkdir pretrained_models\diffusion_prior 2>nul
|
||||||
|
mkdir pretrained_models\strand_prior 2>nul
|
||||||
|
|
||||||
|
REM 下载 diffusion_prior 模型
|
||||||
cd pretrained_models\diffusion_prior
|
cd pretrained_models\diffusion_prior
|
||||||
echo 正在更新Neural Haircut Diffusion Prior模型...
|
echo 正在下载Neural Haircut Diffusion Prior模型...
|
||||||
python -m gdown "%NEURAL_HAIRCUT_MODEL_URL%"
|
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 ..\..
|
cd ..\..
|
||||||
|
|
||||||
REM 下载 PIXIE 文件
|
REM 下载 PIXIE 文件
|
||||||
|
mkdir PIXIE 2>nul
|
||||||
cd PIXIE
|
cd PIXIE
|
||||||
IF EXIST "pixie_data.tar" (
|
IF EXIST "pixie_data.tar" (
|
||||||
echo PIXIE模型已存在,跳过下载...
|
echo PIXIE模型已存在,跳过下载...
|
||||||
|
Loading…
Reference in New Issue
Block a user