Spaces:
Paused
Paused
Abdulrahman1989
commited on
Commit
·
1598c25
1
Parent(s):
c35ebce
fix
Browse files- Image3DProcessor.py +2 -2
Image3DProcessor.py
CHANGED
@@ -75,14 +75,14 @@ class Image3DProcessor:
|
|
75 |
view_to_world_source = view_to_world_source.to(self.device)
|
76 |
rot_transform_quats = rot_transform_quats.to(self.device)
|
77 |
|
78 |
-
reconstruction_unactivated = model(
|
79 |
image_tensor.unsqueeze(0).unsqueeze(0),
|
80 |
view_to_world_source,
|
81 |
rot_transform_quats,
|
82 |
None,
|
83 |
activate_output=False
|
84 |
)
|
85 |
-
|
86 |
reconstruction = {k: v[0].contiguous() for k, v in reconstruction_unactivated.items()}
|
87 |
reconstruction["scaling"] = model.scaling_activation(reconstruction["scaling"])
|
88 |
reconstruction["opacity"] = model.opacity_activation(reconstruction["opacity"])
|
|
|
75 |
view_to_world_source = view_to_world_source.to(self.device)
|
76 |
rot_transform_quats = rot_transform_quats.to(self.device)
|
77 |
|
78 |
+
reconstruction_unactivated = self.model(
|
79 |
image_tensor.unsqueeze(0).unsqueeze(0),
|
80 |
view_to_world_source,
|
81 |
rot_transform_quats,
|
82 |
None,
|
83 |
activate_output=False
|
84 |
)
|
85 |
+
|
86 |
reconstruction = {k: v[0].contiguous() for k, v in reconstruction_unactivated.items()}
|
87 |
reconstruction["scaling"] = model.scaling_activation(reconstruction["scaling"])
|
88 |
reconstruction["opacity"] = model.opacity_activation(reconstruction["opacity"])
|