From 21776459cc83c54d0fc620284fdc38094a80a3b5 Mon Sep 17 00:00:00 2001
From: Nan Huang <hannahuang411@gmail.com>
Date: Sat, 1 Jun 2024 23:05:40 +0800
Subject: [PATCH] update

---
 .DS_Store                | Bin 6148 -> 10244 bytes
 README.md                |   2 +-
 arguments/__init__.py    |   1 +
 arguments/stage2.py      |  10 ++++++++++
 scene/__init__.py        |   3 ++-
 scene/dataset_readers.py |   4 ++--
 6 files changed, 16 insertions(+), 4 deletions(-)
 create mode 100644 arguments/stage2.py

diff --git a/.DS_Store b/.DS_Store
index ac98b5bc0b437e0ea5c4cb195d7cfce09860e0f5..a186833ea59489a013eb328534c24cfd4eefa751 100644
GIT binary patch
literal 10244
zcmeHMO>7%Q6n+ysjh%!NXe;$nX$2P~Lh3|Hp+Z7#>K-b76$cQa;M%cCEY`cmUdNOW
zWUg=l1P3mhKmrbcK%zZ!=m~K`?S&&}!j%I*-<w@~-)we+BZ?YTyVmZrcix+s_kC~P
z{FsQ;O0T|1WD!vgH_N$MJYp^`=X=jJ5xjXEQo)|=jm7f1+iLI8Okbo3qzI%4qzI%4
zqzD`q1Tbgwm7DQp`b-f>5l9gjA;9^;!p*W6$bm0)>cB&90U)z@tuK6^<1dg_Ad7(<
z_!1?!ly3^@o2p=op?q_^w*tLlAP2tm%}E8D4;8emf*p#|t)qTRtCK4FGJU29qzLRq
zfcNfMTA&STk|+1?`IFZkl4UKoRI1r-4exSqesQz*{p9s?W_f+z^7dwUHjuA;15?UV
zl{#p-M=ffQLmqW6+U|PL3J>J3aWDL>FE)s?sP((;z~Wl1Z8y7Kp>RV@WU`Y-rgBrc
zncPd|O}Ah6s$Q?VQtiCRzt&vWt_{CeYtJ<s{mR_&^RDAnn@*#n4Vv`^R4zT=bn0$@
z#qBxuuC^`v1-X1KUzt0(y}evmp0}1BDeTN!+h>;w^VagiXLfe-xfA!Fess0+s@-zj
zcXZ!L1*wbf@_&z>@5Xy3*&sT5HC{{d^7Mnh{&3ajB?jA;8#1)NrsRrfz#WaLZ>!vK
zY}eU9DSLJVY<BT!?B!9NLb>G8(7oLZp3w|0E&fLa4WA@;nYA00C_a+4KX-o`jwYiH
z$ybt&WGeG4T8Z!w;b@xH$;C+90!tn6v_NgTVvae*(I~mfQXR~CR5kU8NB<{cY>xD9
zH=nm?M2`HCToV1x3yx8l<nia_&C{2%K~@YflN`pCOa4e%s5-1W!X3Big1@*N8RbFi
znf*TXOHj8ill@-bcwu)n>~~Ne=8Cv-qu-?aeYeixTbM^WI>v@MeSxkZzuRW?=!_tH
zJxi}(9<cGLpq?#6oShh*((WG2A5~a#=4s-a1N)BdNnOV{;@oVR<5nFNaXUs0CX(~l
zf#eug663EveE+e6kMUTRA^HciG<hkaqclxV&_y~=7w9P}<GBDf>JdyZ&l;G_0{u4d
zYy;J2z~34e?I89xO*>|BJUbJ?6JMfViET*SKL2=x+dldsTI1-)V#Lwv!N(?#u%yT{
ze7zUwaHR2*$4o=VU<K$_QHv8-hmO6sMC|2h6^xvt5<Qu~3};!HjpuZmuOZ@joHfFg
z6Wa6Wb@EJxcm<_oi<9Uj_tl|>sj0b4(7Z!3y71kI$4rKoqoPzsVBzQo!;x!fgfSPd
zVMJNxG8&LAL~{F-W`(1#4HhFAo@Zo(!-)1B!cN31gCY9pJKz`5kSu+h$-f=Uy?eAm
zrjoS`wv=WoQcr}jX)xBK8pfZCd6{RfMwB-_+J6(HJ=RA0sTS}15q`$%NpJt^e>#lG
z1l`c*f76woD_eV>)0r?ii*ZhoeoexE53y@U5zsj$K~wro$97xrTht@H{bT2|p|^%c
z=wrNw^o)+bhX&{v^^u)<J>p`W5SPCbR>X$CGxUZYsTHhQ*y{#zZ#OzOGW!!R3--DF
z!*yl+?pQOB7n1+~KU|$ody^uNB5+F)P}x$sw1O&CLbZI>ZsER+n=jTI_)?qT!7BV-
zAIIYdAICosyeqKe3+2T?4t%N2@m`xlYI8{cH2;75Ghmz(xzQfqn5okJzdj!@=2o(p
d%yj=BId_n2n1k)&<*CuZEKOdD%uB-l{~zz6iXs32

delta 259
zcmZn(XfcprU|?W$DortDU=RQ@Ie-{MGjUE#6q~50D9Qqo2aA<36epDz7bNB6Cv8kz
z&dA6zxlTxjm7SfPn|*SiRG2I$5X1{eR9BmrS?DO3m|N89C{$Y-8R#gO8k$Z{5Hy~A
zS6GIVg&~C@pCOr{c(Q?%3^v_nmZm`628M9mCYDAZ-9VQ>b(RGe<>ln(rGs3yF;SUi
vF*^r`ATy8$1OnVZ!WHEHjfvlxC-bW~f;<TF0mvf^Ob{P}#Wu(D%wYxq0}d|!

diff --git a/README.md b/README.md
index 8e45b8c..7145b18 100644
--- a/README.md
+++ b/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:
 
diff --git a/arguments/__init__.py b/arguments/__init__.py
index 8c937e0..e8bf58b 100644
--- a/arguments/__init__.py
+++ b/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
diff --git a/arguments/stage2.py b/arguments/stage2.py
new file mode 100644
index 0000000..9c1fb8c
--- /dev/null
+++ b/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,
+)
\ No newline at end of file
diff --git a/scene/__init__.py b/scene/__init__.py
index b7626ce..0508091 100644
--- a/scene/__init__.py
+++ b/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:
diff --git a/scene/dataset_readers.py b/scene/dataset_readers.py
index 36c0c52..7e028b5 100644
--- a/scene/dataset_readers.py
+++ b/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 = {}