Fix rotation: start at 330° (3rd UNLOCKED view), fixed 20° pitch, r=2 fov=36

trellis2/utils/render_utils.py

@@ -77,9 +77,15 @@ def render_frames(sample, extrinsics, intrinsics, options={}, verbose=True, **kw
     return rets
 
 
-def render_video(sample, resolution=1024, bg_color=(0, 0, 0), num_frames=120, r=2, fov=40, **kwargs):
-    yaws = -torch.linspace(0, 2 * 3.1415, num_frames) + np.pi/2
-    pitch = 0.25 + 0.5 * torch.sin(torch.linspace(0, 2 * 3.1415, num_frames))
+def render_video(sample, resolution=1024, bg_color=(0, 0, 0), num_frames=120, r=2, fov=36, **kwargs):
+    # Match UNLOCKED render_snapshot exactly:
+    # - offset = (5π/6, 20°) means views at 150°, 240°, 330°, 60°
+    # - 3rd view is 330° = start point for video
+    # - Pitch = 20 degrees (fixed)
+    start_yaw = 330 / 180 * np.pi  # 3rd view from UNLOCKED (330 degrees)
+    pitch_val = 20 / 180 * np.pi   # Fixed pitch from UNLOCKED offset
+    yaws = torch.linspace(start_yaw, 2 * np.pi + start_yaw, num_frames)
+    pitch = torch.tensor([pitch_val] * num_frames)
     yaws = yaws.tolist()
     pitch = pitch.tolist()
     extrinsics, intrinsics = yaw_pitch_r_fov_to_extrinsics_intrinsics(yaws, pitch, r, fov)