Add custom datasets

opt/configs/custom.json

@@ -0,0 +1,23 @@
+{
+    "reso": "[[128, 128, 128], [256, 256, 256], [512, 512, 512]]",
+    "scale": 1.0,
+    "n_iters": 102400,
+    "background_nlayers": 64,
+    "background_reso": 1024,
+    "cam_scale_factor": 0.95,
+    "upsamp_every": 25600,
+    "lr_sigma": 3e1,
+    "lr_sh": 1e-2,
+    "lr_sigma_delay_steps": 0,
+    "lr_fg_begin_step": 1000,
+    "thresh_type": "weight",
+    "weight_thresh": 1.28,
+    "lambda_tv": 5e-5,
+    "lambda_tv_sh": 5e-3,
+    "lambda_tv_background_sigma": 1e-3,
+    "lambda_tv_background_color": 1e-3,
+    "lambda_beta": 1e-5,
+    "lambda_sparsity": 1e-11,
+    "background_brightness": 1.0,
+    "tv_early_only": 0
+}

opt/configs/custom_iphone.json

@@ -0,0 +1,24 @@
+{
+    "reso": "[[128, 128, 128], [256, 256, 256], [512, 512, 512]]",
+    "scale": 1.0,
+    "n_iters": 102400,
+    "cam_scale_factor": 1.5,
+    "background_nlayers": 64,
+    "background_reso": 1024,
+    "cam_scale_factor": 0.95,
+    "upsamp_every": 25600,
+    "lr_sigma": 3e1,
+    "lr_sh": 1e-2,
+    "lr_sigma_delay_steps": 35000,
+    "lr_fg_begin_step": 10000,
+    "thresh_type": "weight",
+    "weight_thresh": 1.28,
+    "lambda_tv": 5e-5,
+    "lambda_tv_sh": 5e-3,
+    "lambda_tv_background_sigma": 1e-3,
+    "lambda_tv_background_color": 1e-3,
+    "lambda_beta": 1e-5,
+    "lambda_sparsity": 1e-11,
+    "background_brightness": 1.0,
+    "tv_early_only": 0
+}

opt/scripts/view_data.py

@@ -286,7 +286,7 @@ def get_transform(c2w):
 
         # (3) Rescale the scene using camera distances
         scale = 1.0 / np.median(np.linalg.norm(t + translate, axis=-1))
-        scale *= 1.5
+        scale *= 0.95
         return transform, scale
 
     T, scale = get_transform(all_poses)