Update trellis/utils/render_utils.py

trellis/utils/render_utils.py

@@ -94,9 +94,11 @@ def render_video(sample, resolution=512, bg_color=(0, 0, 0), num_frames=300, r=2
     
     # Use a more consistent isometric pitch angle (around 35.26 degrees)
     isometric_pitch = 0.615  # approximately 35.26 degrees in radians for true isometric
-    pitch = [isometric_pitch + 0.1 * torch.sin(2 * 3.1415 * i / num_frames) for i in range(num_frames)]
+    pitch_variation = torch.linspace(0, 2 * 3.1415, num_frames)
+    pitch = isometric_pitch + 0.1 * torch.sin(pitch_variation)
     
     yaws = yaws.tolist()
+    pitch = pitch.tolist()
     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)