Update trellis/utils/render_utils.py

trellis/utils/render_utils.py

@@ -31,13 +31,7 @@ def yaw_pitch_r_fov_to_extrinsics_intrinsics(yaws, pitchs, rs, fovs):
             torch.sin(pitch),
         ]).cuda() * r
         extr = utils3d.torch.extrinsics_look_at(orig, torch.tensor([0, 0, 0]).float().cuda(), torch.tensor([0, 0, 1]).float().cuda())
-        # Create intrinsics matrix manually to avoid API issues
-        focal_length = 1.0 / (2.0 * torch.tan(fov / 2.0))
-        intr = torch.tensor([
-            [focal_length, 0, 0.5],
-            [0, focal_length, 0.5],
-            [0, 0, 1]
-        ], dtype=torch.float32, device='cuda')
+        intr = utils3d.torch.intrinsics_from_fov(fov, fov, fov, fov, 1.0)
         extrinsics.append(extr)
         intrinsics.append(intr)
     if not is_list:
@@ -94,10 +88,9 @@ def render_frames(sample, extrinsics, intrinsics, options={}, colors_overwrite=N
 
 
 def render_video(sample, resolution=512, bg_color=(0, 0, 0), num_frames=300, r=2, fov=40, **kwargs):
-    # Start with isometric view (45 degrees yaw offset)
-    isometric_yaw_offset = 3.1415 / 4  # 45 degrees in radians
-    yaws = torch.linspace(isometric_yaw_offset, 2 * 3.1415 + isometric_yaw_offset, num_frames)
-    pitch = 0.25 + 0.5 * torch.sin(torch.linspace(0, 2 * 3.1415, num_frames))
+    # Isometric-like camera angle with Y-axis only rotation
+    yaws = torch.linspace(0, 2 * 3.1415, num_frames)  # Full rotation around Y-axis
+    pitch = torch.tensor([20.0 * 3.1415 / 180.0] * num_frames)  # Fixed pitch at 20 degrees (isometric-like)
     yaws = yaws.tolist()
     pitch = pitch.tolist()
     extrinsics, intrinsics = yaw_pitch_r_fov_to_extrinsics_intrinsics(yaws, pitch, r, fov)