bugfix github.com/google/hypernerf/issues/24

treder · Apr 15, 2022 · e97b0e2 · e97b0e2
1 parent 6cae814
commit e97b0e2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nerfies/model_utils.py b/nerfies/model_utils.py
@@ -104,7 +104,7 @@ def volumetric_rendering(rgb,
   last_sample_z = 1e10 if sample_at_infinity else 1e-19
   dists = jnp.concatenate([
       z_vals[..., 1:] - z_vals[..., :-1],
-      jnp.broadcast_to([last_sample_z], z_vals[..., :1].shape)
+      jnp.broadcast_to(jnp.array([last_sample_z]), z_vals[..., :1].shape)
   ], -1)
   dists = dists * jnp.linalg.norm(dirs[..., None, :], axis=-1)
   alpha = 1.0 - jnp.exp(-sigma * dists)