update

README.md

@@ -64,7 +64,7 @@ If you want to try novel view  synthesis, use
 For training next clip (eg. 51-100 frames), run 
 ```
 python train.py -s $data_dir --port 6017 --expname "waymo" --model_path $model_path 
---prior_checkpoint "$prior_dir/chkpnt_fine_50000.pth"
+--prior_checkpoint "$prior_dir/chkpnt_fine_50000.pth" --configs "arguments/stage2.py"
 ```
 Also, you can load an existing checkpoint with:
 

arguments/__init__.py

@@ -62,6 +62,7 @@ def __init__(self, parser, sentinel=False):
         # waymo
         self.start_time = 0 # now hard-coded
         self.end_time = 49
+        self.original_start_time = 0 # now hard-coded
         self.num_objs = 256 
         self.num_pts = 1500000 
         # mask loading options

arguments/stage2.py

@@ -0,0 +1,10 @@
+ModelParams = dict(
+    stride = 0,
+    original_start_time = 0,
+    start_time = 50,
+    end_time = 99,
+)
+OptimizationParams = dict(
+    coarse_iterations = 5000,
+    iterations = 50000,
+)

scene/__init__.py

@@ -74,7 +74,8 @@ def __init__(self, args : ModelParams, gaussians : GaussianModel, load_iteration
                                     save_occ_grid = args.save_occ_grid,
                                     occ_voxel_size = args.occ_voxel_size,
                                     recompute_occ_grid = args.recompute_occ_grid,
-                                    stride = args.stride
+                                    stride = args.stride,
+                                    original_start_time = args.original_start_time,
                                     )
             dataset_type="waymo"
         else:

scene/dataset_readers.py

@@ -600,7 +600,7 @@ def readWaymoInfo(path, white_background, eval, extension=".png", use_bg_gs=Fals
                   load_intrinsic = False, load_c2w = False,
                   start_time = 0, end_time = -1, num_pts = 5000, 
                   save_occ_grid = False, occ_voxel_size = 0.4, recompute_occ_grid=True,
-                  stride = 10
+                  stride = 10 , original_start_time = 0
                   ):
     ORIGINAL_SIZE = [[1280, 1920], [1280, 1920], [1280, 1920], [884, 1920], [884, 1920]]
     OPENCV2DATASET = np.array(
@@ -675,7 +675,7 @@ def readWaymoInfo(path, white_background, eval, extension=".png", use_bg_gs=Fals
     # ------------------
     # construct timestamps
     # ------------------
-    original_start_time = 0
+    # original_start_time = 0
     idx_list = range(original_start_time, end_time)
     # map time to [0,1]
     timestamp_mapper = {}