File size: 1,431 Bytes
cc9780d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 |
model:
ae: #ae model is loaded to
type: TriVAE
point_emb_dim: 48
padding: 0.1
encoder:
plane_reso: 128
plane_latent_dim: 32
latent_dim: 32
unet:
depth: 4
merge_mode: concat
start_filts: 32
output_dim: 64
decoder:
plane_reso: 128
latent_dim: 32
n_blocks: 5
query_emb_dim: 48
hidden_dim: 128
unet:
depth: 4
merge_mode: concat
start_filts: 64
output_dim: 32
dm:
type: triplane_diff_multiimg_cond
backbone: resunet_multiimg_direct_atten
diff_reso: 64
input_channel: 32
output_channel: 32
triplane_padding: 0.1 #should be consistent with padding in ae
use_par: True
par_channel: 32
par_emb_dim: 48
norm: "batch"
img_in_channels: 1280
vit_reso: 16
use_cat_embedding: ???
block_type: multiview_local
par_point_encoder:
plane_reso: 64
plane_latent_dim: 32
n_blocks: 5
unet:
depth: 3
merge_mode: concat
start_filts: 32
output_dim: 32
criterion:
type: EDMLoss_MultiImgCond
use_par: True
dataset:
type: Occ_Par_MultiImg
data_path: ???
surface_size: 20000
par_pc_size: 2048
load_proj_mat: True
load_image: True
par_point_aug: 0.5
par_prefix: "aug7_" # prefix of the filenames of the partial point cloud
jitter_partial_train: 0.02
jitter_partial_val: 0.0
|