configs/icon-filter.yaml

name: icon-filter
ckpt_dir: "./data/ckpt/"
resume_path: "https://huggingface.co/Yuliang/ICON/resolve/main/icon-filter.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: "icon" # icon/pamir/icon
  use_filter: True
  in_geo: (('normal_F',3), ('normal_B',3))
  in_nml: (('image',3), ('T_normal_F',3), ('T_normal_B',3))
  smpl_feats: ['sdf', 'norm', 'vis', 'cmap']
  gtype: 'HGPIFuNet'
  norm_mlp: 'batch'
  hourglass_dim: 6
  smpl_dim: 7

# user defined
mcube_res: 512    # occupancy field resolution, higher --> more details
clean_mesh: False # if True, will remove floating pieces