Update trellis/utils/render_utils.py

trellis/utils/render_utils.py

@@ -31,7 +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())
-        intr = utils3d.torch.intrinsics_from_fov(fov, fov)
+        intr = utils3d.torch.intrinsics_from_fov_xy(fov, fov)
         extrinsics.append(extr)
         intrinsics.append(intr)
     if not is_list:
@@ -88,14 +88,15 @@ 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):
-    # Match the model viewer camera angle: yaw=45deg, pitch=75deg
-    # Convert degrees to radians for the camera calculations
+    # Match the model viewer camera angle: isometric-like view
+    # Model viewer uses camera-orbit="45deg 75deg 100%" but that's spherical coords
+    # For render coordinates, we need a lower pitch angle for isometric view
     yaws = torch.linspace(0, 2 * 3.1415, num_frames)  # Full rotation around Y-axis
-    pitch = torch.tensor([75.0 * 3.1415 / 180.0] * num_frames)  # Fixed pitch at 75 degrees
+    pitch = torch.tensor([20.0 * 3.1415 / 180.0] * num_frames)  # Fixed pitch at 20 degrees (isometric-like)
     yaws = yaws.tolist()
     pitch = pitch.tolist()
-    # Adjust distance to match model viewer's final position (closer view)
-    r = 1.5  # Closer than default 2, similar to model viewer's 100% distance
+    # Use original distance to keep model in bounds
+    r = 2.0  # Back to original distance to prevent clipping
     extrinsics, intrinsics = yaw_pitch_r_fov_to_extrinsics_intrinsics(yaws, pitch, r, fov)
     return render_frames(sample, extrinsics, intrinsics, {'resolution': resolution, 'bg_color': bg_color}, **kwargs)