Spaces:
Paused
Paused
Commit ·
91765b3
1
Parent(s): aafb120
fix self device
Browse files- Image3DProcessor.py +3 -3
Image3DProcessor.py
CHANGED
|
@@ -96,9 +96,9 @@ class Image3DProcessor:
|
|
| 96 |
for r_idx in range(world_view_transforms.shape[0]):
|
| 97 |
rendered_image = render_predicted(
|
| 98 |
reconstruction,
|
| 99 |
-
world_view_transforms[r_idx].to(device),
|
| 100 |
-
full_proj_transforms[r_idx].to(device),
|
| 101 |
-
camera_centers[r_idx].to(device),
|
| 102 |
background,
|
| 103 |
model_cfg,
|
| 104 |
focals_pixels=None
|
|
|
|
| 96 |
for r_idx in range(world_view_transforms.shape[0]):
|
| 97 |
rendered_image = render_predicted(
|
| 98 |
reconstruction,
|
| 99 |
+
world_view_transforms[r_idx].to(self.device),
|
| 100 |
+
full_proj_transforms[r_idx].to(self.device),
|
| 101 |
+
camera_centers[r_idx].to(self.device),
|
| 102 |
background,
|
| 103 |
model_cfg,
|
| 104 |
focals_pixels=None
|