add opacity threshold

flash3d/util/vis3d.py

@@ -27,10 +27,19 @@ def export_ply(
     scales: Float[Tensor, "gaussian 3"],
     rotations: Float[Tensor, "gaussian 4"],
     harmonics: Float[Tensor, "gaussian 3 d_sh"],
-    opacities: Float[Tensor, "gaussian"],
+    opacities: Float[Tensor, "gaussian 1"],
     path: Path,
 ):
     path = Path(path)
+
+    min_opacity = 0.50
+    valid = opacities[..., 0] > min_opacity
+    means = means[valid, ...]
+    scales = scales[valid, ...]
+    rotations = rotations[valid, ...]
+    harmonics = harmonics[valid, ...]
+    opacities = opacities[valid, ...]
+
     # Shift the scene so that the median Gaussian is at the origin.
     means = means - means.median(dim=0).values