|
from yacs.config import CfgNode as CN |
|
|
|
_C = CN() |
|
_C.seed = 0 |
|
_C.output_dir = "results" |
|
_C.result_name = "test_all" |
|
|
|
_C.triplet_sampling = "random" |
|
_C.load_original_mesh = False |
|
|
|
_C.num_pos = 64 |
|
_C.num_neg_random = 256 |
|
_C.num_neg_hard_pc = 128 |
|
_C.num_neg_hard_emb = 128 |
|
|
|
_C.vertex_feature = False |
|
_C.n_point_per_face = 2000 |
|
_C.n_sample_each = 10000 |
|
_C.preprocess_mesh = False |
|
|
|
_C.regress_2d_feat = False |
|
|
|
_C.is_pc = False |
|
|
|
_C.cut_manifold = False |
|
_C.remesh_demo = False |
|
_C.correspondence_demo = False |
|
|
|
_C.save_every_epoch = 10 |
|
_C.training_epochs = 30 |
|
_C.continue_training = False |
|
|
|
_C.continue_ckpt = None |
|
_C.epoch_selected = "epoch=50.ckpt" |
|
|
|
_C.triplane_resolution = 128 |
|
_C.triplane_channels_low = 128 |
|
_C.triplane_channels_high = 512 |
|
_C.lr = 1e-3 |
|
_C.train = True |
|
_C.test = False |
|
|
|
_C.inference_save_pred_sdf_to_mesh=True |
|
_C.inference_save_feat_pca=True |
|
_C.name = "test" |
|
_C.test_subset = False |
|
_C.test_corres = False |
|
_C.test_partobjaversetiny = False |
|
|
|
_C.dataset = CN() |
|
_C.dataset.type = "Demo_Dataset" |
|
_C.dataset.data_path = "objaverse_data/" |
|
_C.dataset.train_num_workers = 64 |
|
_C.dataset.val_num_workers = 32 |
|
_C.dataset.train_batch_size = 2 |
|
_C.dataset.val_batch_size = 2 |
|
_C.dataset.all_files = [] |
|
|
|
_C.voxel2triplane = CN() |
|
_C.voxel2triplane.transformer_dim = 1024 |
|
_C.voxel2triplane.transformer_layers = 6 |
|
_C.voxel2triplane.transformer_heads = 8 |
|
_C.voxel2triplane.triplane_low_res = 32 |
|
_C.voxel2triplane.triplane_high_res = 256 |
|
_C.voxel2triplane.triplane_dim = 64 |
|
_C.voxel2triplane.normalize_vox_feat = False |
|
|
|
|
|
_C.loss = CN() |
|
_C.loss.triplet = 0.0 |
|
_C.loss.sdf = 1.0 |
|
_C.loss.feat = 10.0 |
|
_C.loss.l1 = 0.0 |
|
|
|
_C.use_pvcnn = False |
|
_C.use_pvcnnonly = True |
|
|
|
_C.pvcnn = CN() |
|
_C.pvcnn.point_encoder_type = 'pvcnn' |
|
_C.pvcnn.use_point_scatter = True |
|
_C.pvcnn.z_triplane_channels = 64 |
|
_C.pvcnn.z_triplane_resolution = 256 |
|
_C.pvcnn.unet_cfg = CN() |
|
_C.pvcnn.unet_cfg.depth = 3 |
|
_C.pvcnn.unet_cfg.enabled = True |
|
_C.pvcnn.unet_cfg.rolled = True |
|
_C.pvcnn.unet_cfg.use_3d_aware = True |
|
_C.pvcnn.unet_cfg.start_hidden_channels = 32 |
|
_C.pvcnn.unet_cfg.use_initial_conv = False |
|
|
|
_C.use_2d_feat = False |
|
_C.inference_metrics_only = False |
|
|