cull points

app.py

@@ -114,10 +114,8 @@ def main():
         loop_out_path = os.path.join(os.path.dirname(ply_out_path), "loop.mp4")
         imageio.mimsave(loop_out_path, loop_renders, fps=25)"""
         # export reconstruction to ply
-        print("exporting")
         export_to_obj(reconstruction_unactivated, ply_out_path)
-        print("exported")
-        print(os.listdir(os.path.dirname(os.path.abspath(__file__))))
+
         return ply_out_path
 
     with gr.Blocks() as demo:

utils/app_utils.py

@@ -158,13 +158,15 @@ def export_to_obj(reconstruction, ply_out_path):
         assert v.shape[0] == 1, "Expected batch size to be 0"
         reconstruction[k] = v[0]
 
-    xyz = reconstruction["xyz"].detach().cpu().numpy()
+    non_transparent_points = torch.where(reconstruction["opacity"] > 0.005)[0]
+
+    xyz = reconstruction["xyz"][non_transparent_points].detach().cpu().numpy()
     normals = np.zeros_like(xyz)
-    f_dc = reconstruction["features_dc"].detach().transpose(1, 2).flatten(start_dim=1).contiguous().cpu().numpy()
-    f_rest = reconstruction["features_rest"].detach().transpose(1, 2).flatten(start_dim=1).contiguous().cpu().numpy()
-    opacities = reconstruction["opacity"].detach().cpu().numpy()
-    scale = reconstruction["scaling"].detach().cpu().numpy()
-    rotation = reconstruction["rotation"].detach().cpu().numpy()
+    f_dc = reconstruction["features_dc"][non_transparent_points].detach().transpose(1, 2).flatten(start_dim=1).contiguous().cpu().numpy()
+    f_rest = reconstruction["features_rest"][non_transparent_points].detach().transpose(1, 2).flatten(start_dim=1).contiguous().cpu().numpy()
+    opacities = reconstruction["opacity"][non_transparent_points].detach().cpu().numpy()
+    scale = reconstruction["scaling"][non_transparent_points].detach().cpu().numpy()
+    rotation = reconstruction["rotation"][non_transparent_points].detach().cpu().numpy()
 
     dtype_full = [(attribute, 'f4') for attribute in construct_list_of_attributes()]