Update trellis/utils/render_utils.py

trellis/utils/render_utils.py

@@ -88,10 +88,14 @@ 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):
-    yaws = torch.linspace(0, 2 * 3.1415, num_frames)
-    pitch = 0.25 + 0.5 * torch.sin(torch.linspace(0, 2 * 3.1415, num_frames))
+    # Match the model viewer camera angle: yaw=45deg, pitch=75deg
+    # Convert degrees to radians for the camera calculations
+    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
     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
     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)