commit DM-NeRF train in replica and scannet for sparse exp

config.py

@@ -24,6 +24,10 @@ def config_parser():
                         help='layers in network')
     parser.add_argument("--netwidth", type=int, default=256,
                         help='channels per layer')
+
+    parser.add_argument("--label_sparse_inv", type=int, default=1,
+                        help='label sparse inv')
+
     # 32*32*4
     parser.add_argument("--N_train", type=int, default=4096,
                         help='batch size (number of random rays per gradient step)')
@@ -77,7 +81,7 @@ def config_parser():
                         help='frequency of console printout and metric loggin')
     parser.add_argument("--i_img", type=int, default=500,
                         help='frequency of tensorboard image logging')
-    parser.add_argument("--i_save", type=int, default=10000,
+    parser.add_argument("--i_save", type=int, default=50000,
                         help='frequency of weight ckpt saving')
     parser.add_argument("--i_test", type=int, default=50000,
                         help='frequency of testset saving')
@@ -142,6 +146,19 @@ def initial():
     parser = config_parser()
     args = parser.parse_args()
 
+    sparse_inv = args.label_sparse_inv
+    if args.expname[-4:] != 'full':
+        expname_list = args.expname.split('_')
+        num = int(args.expname[-2:])
+        try:
+            flag = (int(expname_list[-1]) == sparse_inv) or (num == sparse_inv)
+        except:
+            flag = (num == sparse_inv)
+
+        if not flag:
+            exp_str = '-new_sparse' + str(sparse_inv).zfill(2)
+            args.expname += exp_str
+
     # get log time
     if args.log_time is None:
         args.log_time = time.strftime("%Y%m%d%H%M", time.localtime())

configs/replica/train/m40/office_2_sparse_inv.txt

@@ -0,0 +1,31 @@
+expname = office_2_sparse10
+# sparse inv
+label_sparse_inv = 10
+
+basedir = ./logs/replica
+# 2080ti
+# datadir = /data/dzy_data/nerf/datasets/nerf_replica/replica_ins/office_2
+# audodl-3090
+# datadir = /root/autodl-tmp/datasets/nerf_replica/replica_ins/office_2
+# m40
+datadir = /media/alpha4TB/exchange/nerf/DM-NeRF/replica/replica/office_2
+
+
+lrate_decay = 500
+
+N_samples = 64
+N_importance = 128
+N_train = 1024
+N_test = 1024
+
+
+near = 0.0
+far = 5.7
+testskip = 1
+
+penalize
+tolerance = 0.05
+deta_w = 0.05
+
+# render test step
+i_test = -1

configs/replica/train/office_0.txt

@@ -1,12 +1,13 @@
 expname = office_0
 basedir = ./logs/replica
-datadir = ./data/replica/office_0
+# 2080ti
+datadir = /data/dzy_data/nerf/datasets/nerf_replica/replica_ins/office_0
 
 lrate_decay = 500
 
 N_samples = 64
 N_importance = 128
-N_train = 3072
+N_train = 1024
 N_test = 4096
 
 

configs/replica/train/office_0_full.txt

@@ -0,0 +1,27 @@
+expname = office_0_full
+
+label_sparse_inv = 1
+
+basedir = ./logs/replica
+# 2080ti
+# datadir = /data/dzy_data/nerf/datasets/nerf_replica/replica_ins/office_0
+# audodl-3090
+datadir = /root/autodl-tmp/datasets/nerf_replica/replica_ins/office_0
+
+lrate_decay = 500
+
+N_samples = 64
+N_importance = 128
+N_train = 1500
+N_test = 1500
+
+near = 0.0
+far = 4.7
+testskip = 1
+
+penalize
+tolerance = 0.05
+deta_w = 0.05
+
+# render test step
+i_test = -1

configs/replica/train/office_0_sparse_inv.txt

@@ -0,0 +1,30 @@
+
+expname = office_0_sparse10
+# sparse inv
+label_sparse_inv = 10
+
+
+basedir = ./logs/replica
+# 2080ti
+# datadir = /data/dzy_data/nerf/datasets/nerf_replica/replica_ins/office_0
+# audodl-3090
+datadir = /root/autodl-tmp/datasets/nerf_replica/replica_ins/office_0
+
+lrate_decay = 500
+
+N_samples = 64
+N_importance = 128
+N_train = 1500
+N_test = 1500
+
+
+near = 0.0
+far = 4.7
+testskip = 1
+
+penalize
+tolerance = 0.05
+deta_w = 0.05
+
+# render test step
+i_test = -1

configs/replica/train/office_2_full.txt

@@ -0,0 +1,28 @@
+expname = office_2_full
+
+label_sparse_inv = 1
+
+basedir = ./logs/replica
+# 2080ti
+# datadir = /data/dzy_data/nerf/datasets/nerf_replica/replica_ins/office_2
+# audodl-3090
+datadir = /root/autodl-tmp/datasets/nerf_replica/replica_ins/office_2
+
+lrate_decay = 500
+
+N_samples = 64
+N_importance = 128
+N_train = 1024
+N_test = 1024
+
+
+near = 0.0
+far = 5.7
+testskip = 1
+
+penalize
+tolerance = 0.05
+deta_w = 0.05
+
+# render test step
+i_test = -1

configs/replica/train/office_2_sparse_inv.txt

@@ -0,0 +1,28 @@
+expname = office_2_sparse10
+# sparse inv
+label_sparse_inv = 10
+
+basedir = ./logs/replica
+# 2080ti
+# datadir = /data/dzy_data/nerf/datasets/nerf_replica/replica_ins/office_2
+# audodl-3090
+datadir = /root/autodl-tmp/datasets/nerf_replica/replica_ins/office_2
+
+lrate_decay = 500
+
+N_samples = 64
+N_importance = 128
+N_train = 1024
+N_test = 1024
+
+
+near = 0.0
+far = 5.7
+testskip = 1
+
+penalize
+tolerance = 0.05
+deta_w = 0.05
+
+# render test step
+i_test = -1

configs/replica/train/office_3.txt

@@ -1,19 +1,25 @@
 expname = office_3
 basedir = ./logs/replica
-datadir = ./data/replica/office_3
+# 2080ti
+datadir = /data/dzy_data/nerf/datasets/nerf_replica/replica_ins/office_3
 
 lrate_decay = 500
 
 N_samples = 64
 N_importance = 128
-N_train = 3072
+N_train = 1024
 N_test = 4096
 
 
 near = 0.0
 far = 6.6
-testskip = 4
+testskip = 1
 
 penalize
 tolerance = 0.05
 deta_w = 0.05
+
+# render test step
+i_test = -1
+
+label_sparse_inv = 1

configs/replica/train/office_3_full.txt

@@ -0,0 +1,28 @@
+expname = office_3_full
+# sparse inv
+label_sparse_inv = 1
+
+basedir = ./logs/replica
+# 2080ti
+datadir = /data/dzy_data/nerf/datasets/nerf_replica/replica_ins/office_3
+# audodl-3090
+datadir = /root/autodl-tmp/datasets/nerf_replica/replica_ins/office_3
+
+lrate_decay = 500
+
+N_samples = 64
+N_importance = 128
+N_train = 2048
+N_test = 2048
+
+
+near = 0.0
+far = 6.6
+testskip = 1
+
+penalize
+tolerance = 0.05
+deta_w = 0.05
+
+# render test step
+i_test = -1

configs/replica/train/office_3_sparse_inv.txt

@@ -0,0 +1,30 @@
+expname = office_3_sparse10
+# sparse inv
+label_sparse_inv = 10
+
+basedir = ./logs/replica
+# audodl-3090
+datadir = /root/autodl-tmp/datasets/nerf_replica/replica_ins/office_3
+# 2080ti
+# datadir = /data/dzy_data/nerf/datasets/nerf_replica/replica_ins/office_3
+
+
+lrate_decay = 500
+
+N_samples = 64
+N_importance = 128
+N_train = 1024
+N_test = 1024
+
+
+near = 0.0
+far = 6.6
+testskip = 1
+
+penalize
+tolerance = 0.05
+deta_w = 0.05
+
+# render test step
+i_test = -1
+

configs/replica/train/office_4_full.txt

@@ -0,0 +1,27 @@
+expname = office_4_full
+
+label_sparse_inv = 1
+
+basedir = ./logs/replica
+# 2080ti
+# datadir = /data/dzy_data/nerf/datasets/nerf_replica/replica_ins/office_4
+# audodl-3090
+datadir = /root/autodl-tmp/datasets/nerf_replica/replica_ins/office_4
+
+lrate_decay = 500
+
+N_samples = 64
+N_importance = 128
+N_train = 1500
+N_test = 1500
+
+near = 0.0
+far = 6.2
+testskip = 1
+
+penalize
+tolerance = 0.05
+deta_w = 0.05
+
+# render test step
+i_test = -1

configs/replica/train/office_4_sparse_inv.txt

@@ -0,0 +1,27 @@
+expname = office_4_sparse10
+
+label_sparse_inv = 10
+
+basedir = ./logs/replica
+# 2080ti
+# datadir = /data/dzy_data/nerf/datasets/nerf_replica/replica_ins/office_4
+# audodl-3090
+datadir = /root/autodl-tmp/datasets/nerf_replica/replica_ins/office_4
+
+lrate_decay = 500
+
+N_samples = 64
+N_importance = 128
+N_train = 1500
+N_test = 1500
+
+near = 0.0
+far = 6.2
+testskip = 1
+
+penalize
+tolerance = 0.05
+deta_w = 0.05
+
+# render test step
+i_test = -1

configs/replica/train/room_0_full.txt

@@ -0,0 +1,27 @@
+expname = room_0_full
+
+label_sparse_inv = 1
+
+basedir = ./logs/replica/room
+# 2080ti
+# datadir = /data/dzy_data/nerf/datasets/nerf_replica/replica_ins/room_0
+# audodl-3090
+datadir = /root/autodl-tmp/datasets/nerf_replica/replica_ins/room_0
+lrate_decay = 500
+
+N_samples = 64
+N_importance = 128
+N_train = 1500
+N_test = 1500
+
+
+near = 0.0
+far = 6.5
+testskip = 1
+
+penalize
+tolerance = 0.05
+deta_w = 0.05
+
+# render test step
+i_test = -1