arabago96 commited on
Commit
a2727bc
·
verified ·
1 Parent(s): ed2cc94

Update trellis/utils/render_utils.py

Browse files
Files changed (1) hide show
  1. trellis/utils/render_utils.py +1 -2
trellis/utils/render_utils.py CHANGED
@@ -91,13 +91,12 @@ def render_video(sample, resolution=512, bg_color=(0, 0, 0), num_frames=300, r=2
91
  # Start with isometric view (45 degrees yaw offset)
92
  isometric_yaw_offset = 3.1415 / 4 # 45 degrees in radians
93
  yaws = torch.linspace(isometric_yaw_offset, 2 * 3.1415 + isometric_yaw_offset, num_frames)
94
- pitch = 0.25 + 0.5 * torch.sin(torch.linspace(0, 2 * 3.1415, num_frames))
95
  yaws = yaws.tolist()
96
  pitch = pitch.tolist()
97
  extrinsics, intrinsics = yaw_pitch_r_fov_to_extrinsics_intrinsics(yaws, pitch, r, fov)
98
  return render_frames(sample, extrinsics, intrinsics, {'resolution': resolution, 'bg_color': bg_color}, **kwargs)
99
 
100
-
101
  def render_multiview(sample, resolution=512, nviews=30):
102
  r = 2
103
  fov = 40
 
91
  # Start with isometric view (45 degrees yaw offset)
92
  isometric_yaw_offset = 3.1415 / 4 # 45 degrees in radians
93
  yaws = torch.linspace(isometric_yaw_offset, 2 * 3.1415 + isometric_yaw_offset, num_frames)
94
+ pitch = [0.25] * num_frames # Fixed pitch for isometric view
95
  yaws = yaws.tolist()
96
  pitch = pitch.tolist()
97
  extrinsics, intrinsics = yaw_pitch_r_fov_to_extrinsics_intrinsics(yaws, pitch, r, fov)
98
  return render_frames(sample, extrinsics, intrinsics, {'resolution': resolution, 'bg_color': bg_color}, **kwargs)
99
 
 
100
  def render_multiview(sample, resolution=512, nviews=30):
101
  r = 2
102
  fov = 40