Update trellis/utils/render_utils.py

trellis/utils/render_utils.py

@@ -88,12 +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):
-    # Start with isometric view (45 degrees yaw offset)
-    isometric_yaw_offset = 3.1415 / 4  # 45 degrees in radians
+    # Start with isometric view facing forward (225 degrees yaw for front-facing isometric)
+    isometric_yaw_offset = 5 * 3.1415 / 4  # 225 degrees in radians (front-facing isometric)
     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))
+    
+    # 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)]
+    
     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)