Spaces:
Running
on
Zero
Running
on
Zero
Update trellis/utils/render_utils.py
Browse files
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
|
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
|