Abdulrahman1989 commited on
Commit
1598c25
·
1 Parent(s): c35ebce
Files changed (1) hide show
  1. 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"])