flash3d / config_re10k_v1.yaml
Charlie0804's picture
Update config_re10k_v1.yaml
002ec65 verified
raw
history blame
1.75 kB
model:
name: unidepth_extension_vit
num_layers_pose: 18
backbone: convnextv2_large
min_depth: 0.1
max_depth: 20
use_stereo: false
frame_ids:
- 0
- -1
- 1
scales:
- 0
- 1
- 2
- 3
gauss_novel_frames:
- -1
- 1
- 2
gaussian_rendering: true
resnet_bn_order: pre_bn
stopgrad_depth: false
stopgrad_pose: false
predict_offset: true
num_ch_dec:
- 32
- 32
- 64
- 128
- 256
unified_decoder: false
shift_rays_half_pixel: forward
no_ssim: false
weights_init: pretrained
pose_model_input: pairs
pose_model_type: separate_resnet
max_sh_degree: 1
depth_type: depth_inc
merged_warped_depth: max
depth_thr: 0.1
depth_scale: 0.1
xyz_scale: 0.2
opacity_scale: 1.0
scale_scale: 0.1
scale_lambda: 0.01
depth_bias: -0.1
xyz_bias: 0.0
opacity_bias: 0.0
scale_bias: 0.02
sh_scale: 1.0
scaled_offset: false
bg_colour:
- 0.5
- 0.5
- 0.5
randomise_bg_colour: true
renderer_w_pose: true
gaussians_per_pixel: 5
one_gauss_decoder: false
num_layers: 50
scale_with_depth: false
splat_branch: resnet
depth_cond: true
dataset:
name: nyuv2
split: original
data_path: /scratch/shared/nfs1/eldar/data/nyuv2
colmap_path: /scratch/shared/nfs1/eldar/data/nyuv2_colmap
color_aug: false
height: 256
width: 384
resize: true
baseline_epsilon: 0.001
max_fov: 100.0
skip_bad_shape: true
near: -1.0
far: -1.0
crop_principal_point: false
subset: -1
pad_border_aug: 325wqwe
dilation: random
max_dilation: 15
unpack_pcl_tar: true
crop_border: true
optimiser:
batch_size: 1
train:
data_augmentation: true
mixed_precision: false
load_weights_folder: null
supervised_depth: false
frame_sampling_method: two_forward_one_back