Spaces:
Paused
Paused
Abdulrahman1989
commited on
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
|