Yongsen Mao
Add VP Hough Transform Gaussian Sphere models
4960dfe
raw
history blame contribute delete
503 Bytes
io:
datadir: dataset/nyu_vp/processed_data_aug
dataset: NYU
focal_length: 1.0
logdir: logs/
num_neighbors: 20
num_nodes: 32768
num_vpts: null
num_workers: 4
percentage: 1
resume_from: logs/211103-130207-nyu
tensorboard_port: 0
validation_debug: 120
model:
backbone: stacked_hourglass
batch_size: 3
depth: 4
lneg: 1.0
lpos: 1.0
num_blocks: 1
num_stacks: 1
optim:
amsgrad: true
lr: 0.0004
lr_decay_epoch: 24
max_epoch: 36
name: Adam
weight_decay: 1.0e-05