|
322 | 322 | }, |
323 | 323 | "outputs": [], |
324 | 324 | "source": [ |
325 | | - "# Initialize \u0026 load per-scene NeRF models.\n", |
| 325 | + "# Initialize & load per-scene NeRF models.\n", |
326 | 326 | "\n", |
327 | 327 | "recovered_nerf_state = semantic_utils.load_all_nerf_variables(\n", |
328 | 328 | " save_dir=params.train.nerf_model_ckpt,\n", |
|
401 | 401 | }, |
402 | 402 | "outputs": [], |
403 | 403 | "source": [ |
404 | | - "# Plot XYZ values where density \u003e min_density for various values of min_density.\n", |
| 404 | + "# Plot XYZ values where density > min_density for various values of min_density.\n", |
405 | 405 | "\n", |
406 | 406 | "def plot_density_coordinates(kept_points, color=None, ax=None):\n", |
407 | 407 | " ii, jj, kk = kept_points[:, 0], kept_points[:, 1], kept_points[:, 2]\n", |
|
440 | 440 | }, |
441 | 441 | "outputs": [], |
442 | 442 | "source": [ |
443 | | - "# Plot XYZ values where density \u003e 0, interactively.\n", |
| 443 | + "# Plot XYZ values where density > 0, interactively.\n", |
444 | 444 | "\n", |
445 | 445 | "def plot_density_coordinates_interactive(kept_points, color=None):\n", |
446 | 446 | " assert len(kept_points)\n", |
|
592 | 592 | "fig = plt.figure(figsize=(len(min_density_values) * 4, 4))\n", |
593 | 593 | "axs = []\n", |
594 | 594 | "for i, min_density in enumerate(min_density_values):\n", |
595 | | - " mask = sigma_values \u003e min_density\n", |
| 595 | + " mask = sigma_values > min_density\n", |
596 | 596 | " kept_points = positions[0][mask[0, :, 0]]\n", |
597 | 597 | " kept_points_color = color_values[0][mask[0, :, 0]]\n", |
598 | 598 | " ax = fig.add_subplot(1, len(min_density_values), i+1, projection='3d')\n", |
|
673 | 673 | " *,\n", |
674 | 674 | " semantic_variables: Tree[jnp.ndarray],\n", |
675 | 675 | " semantic_model: volumetric_semantic_model.VolumetricSemanticModel,\n", |
676 | | - ") -\u003e f32[\"D n k\"]:\n", |
| 676 | + ") -> f32[\"D n k\"]:\n", |
677 | 677 | " \"\"\"Predict semantic logits for a set of 3D points.\n", |
678 | 678 | "\n", |
679 | 679 | " Args:\n", |
|
854 | 854 | "# Visualize semantic model predictions across 3D lattice of points.\n", |
855 | 855 | "\n", |
856 | 856 | "MIN_DENSITY = 16\n", |
857 | | - "mask = sigma_values \u003e MIN_DENSITY\n", |
| 857 | + "mask = sigma_values > MIN_DENSITY\n", |
858 | 858 | "kept_points = positions[0][mask[0, :, 0]]\n", |
859 | 859 | "kept_points_color = semantic_predictions[0][mask[0, :, 0]]\n", |
860 | 860 | "print(kept_points.shape)\n", |
|
913 | 913 | "\n", |
914 | 914 | " points = ray_o + depth * ray_d\n", |
915 | 915 | "\n", |
916 | | - " mask = np.all((points \u003e= -1) \u0026 (points \u003c= 1), axis=-1)\n", |
| 916 | + " mask = np.all((points >= -1) & (points <= 1), axis=-1)\n", |
917 | 917 | "\n", |
918 | 918 | " select_points = points[mask]\n", |
919 | 919 | " select_semantics = semantics[mask]\n", |
|
929 | 929 | }, |
930 | 930 | "outputs": [], |
931 | 931 | "source": [ |
932 | | - "# Construct labeled semantic point cloud from ground truth dataset (i.e. using semantic masks, ray origins \u0026 directions from known cameras, and depth)\n", |
| 932 | + "# Construct labeled semantic point cloud from ground truth dataset (i.e. using semantic masks, ray origins & directions from known cameras, and depth)\n", |
933 | 933 | "\n", |
934 | 934 | "labeled_point_cloud = construct_labeled_point_cloud(examples)" |
935 | 935 | ] |
|
0 commit comments