adding camera offsets values

inference/v2v_data/demo.py

@@ -606,7 +606,43 @@ class GetAnchorVideos:
                 ]
             ).to(opts.device).unsqueeze(0)
         )
-        
+            
+        theta = torch.deg2rad(torch.tensor(0)).to(opts.device)
+        sin_value_x = torch.sin(theta)
+        cos_value_x = torch.cos(theta)
+        rot_mat_x = (
+            torch.tensor(
+                [
+                    [1, 0, 0, 0],
+                    [0, cos_value_x, -sin_value_x, 0],
+                    [0, sin_value_x, cos_value_x, 0],
+                    [0, 0, 0, 1],
+                ]
+            )
+            .unsqueeze(0)
+            .repeat(c2ws.shape[0], 1, 1)
+            .to(opts.device)
+        )
+
+        phi = torch.deg2rad(torch.tensor(0)).to(opts.device)
+        sin_value_y = torch.sin(phi)
+        cos_value_y = torch.cos(phi)
+        rot_mat_y = (
+            torch.tensor(
+                [
+                    [cos_value_y, 0, sin_value_y, 0],
+                    [0, 1, 0, 0],
+                    [-sin_value_y, 0, cos_value_y, 0],
+                    [0, 0, 0, 1],
+                ]
+            )
+            .unsqueeze(0)
+            .repeat(c2ws.shape[0], 1, 1)
+            .to(opts.device)
+        )
+
+        c2ws = torch.matmul(rot_mat_x, c2ws)
+        c2w_init = torch.matmul(rot_mat_y, c2ws)
         # c2ws[:, 2, 3] -= r
         # if x is not None:
         #     c2ws[:, 1, 3] += y