Shape-as-Point/configs/learning_based/outlier/ours_3plane.yaml

55 lines
1.1 KiB
YAML
Raw Normal View History

2021-11-08 10:09:50 +00:00
data:
class: null
data_type: psr_full
input_type: pointcloud
path: data/shapenet_psr
num_gt_points: 10000
num_offset: 5
pointcloud_n: 3000
pointcloud_noise: 0.005
pointcloud_outlier_ratio: 0.5
model:
grid_res: 128 # poisson grid resolution
psr_sigma: 2
psr_tanh: True
normal_normalize: False
predict_normal: True
predict_offset: True
c_dim: 32
s_offset: 0.001
encoder: local_pool_pointnet
encoder_kwargs:
hidden_dim: 32
plane_type: ['xz', 'xy', 'yz']
plane_resolution: 64
unet: True
unet_kwargs:
depth: 4
merge_mode: concat
start_filts: 32
decoder: simple_local
decoder_kwargs:
sample_mode: bilinear # bilinear / nearest
hidden_size: 32
train:
batch_size: 32
lr: 5e-4
out_dir: out/shapenet/outlier_ours_3plane
w_psr: 1
model_selection_metric: psr_l2
print_every: 100
checkpoint_every: 200
validate_every: 5000
backup_every: 10000
total_epochs: 400000
visualize_every: 5000
exp_pcl: True
exp_mesh: True
n_workers: 8
n_workers_val: 0
generation:
exp_gt: False
exp_input: True
psr_resolution: 128
psr_sigma: 2