Update trellis/utils/render_utils.py

trellis/utils/render_utils.py

@@ -88,12 +88,12 @@ 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
-    yaws = torch.linspace(isometric_yaw_offset, 2 * 3.1415 + isometric_yaw_offset, num_frames)
-    pitch = [0.25] * num_frames  # Fixed pitch for isometric view
+    # Start at 45 degrees from front and rotate from there
+    start_angle = 3.1415 / 4  # 45 degrees in radians
+    yaws = torch.linspace(start_angle, 2 * 3.1415 + start_angle, num_frames)  # Full rotation starting from 45 degrees
+    pitch = torch.tensor([0.4] * num_frames)  # Fixed pitch for isometric-like view (slightly above)
     yaws = yaws.tolist()
-    # pitch is already a list, no .tolist() needed
+    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)