fix issues/15, issue/7

Author: ryf1123

code/dataset/dtu_train.py

@@ -260,7 +260,8 @@ def __getitem__(self, idx):
             near_fars.append(self.all_near_fars[index_mat])
             intrinsics.append(self.all_intrinsics[index_mat])
 
-            w2cs.append(self.all_extrinsics[index_mat])
+            # issue/7
+            w2cs.append(self.all_extrinsics[index_mat] @ w2c_ref_inv)
 
             if os.path.exists(depth_filename):  # and i == 0
                 depth_h = self.read_depth(depth_filename)

code/utils/renderer.py

@@ -1,4 +1,5 @@
 import torch
+import torch.nn.functional as F
 
 class VolumeRenderer():
     def __init__(self, args=None):
@@ -26,7 +27,10 @@ def render(self, z_val, radiance, geo_value, cos_anneal_ratio=1.0, deviation_net
         inv_s = inv_s0.expand(batch_size, n_samples)
 
         true_cos = -1.0
-        iter_cos = -(-true_cos * 0.5 + 0.5 * (1.0 - cos_anneal_ratio) -true_cos * cos_anneal_ratio)  
+        
+        # FIX: https://github.com/IVRL/VolRecon/issues/15
+        iter_cos = -(F.relu(-true_cos * 0.5 + 0.5) * (1.0 - cos_anneal_ratio) + F.relu(-true_cos) * cos_anneal_ratio)
+        # iter_cos = -(-true_cos * 0.5 + 0.5 * (1.0 - cos_anneal_ratio) -true_cos * cos_anneal_ratio)  
 
         estimated_next_srdf = srdf + iter_cos * interval * 0.5
         estimated_prev_srdf = srdf - iter_cos * interval * 0.5