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
|