ICON / configs /pifu.yaml
Yuliang's picture
done
2d5f249
raw
history blame
675 Bytes
name: pifu
ckpt_dir: "./data/ckpt/"
resume_path: "https://huggingface.co/Yuliang/ICON/resolve/main/pifu.ckpt"
normal_path: "https://huggingface.co/Yuliang/ICON/resolve/main/normal.ckpt"
test_mode: True
batch_size: 1
net:
mlp_dim: [256, 512, 256, 128, 1]
res_layers: [2,3,4]
num_stack: 2
prior_type: "pifu" # icon/pamir/icon
use_filter: True
in_geo: (('image',3), ('normal_F',3), ('normal_B',3))
in_nml: (('image',3), ('T_normal_F',3), ('T_normal_B',3))
gtype: 'HGPIFuNet'
norm_mlp: 'batch'
hourglass_dim: 12
# user defined
mcube_res: 512 # occupancy field resolution, higher --> more details
clean_mesh: False # if True, will remove floating pieces