API Documentation

• pytorch3d.structures
  • join_meshes_as_batch()
  • join_meshes_as_scene()
  • Meshes
    • Meshes.__init__()
    • Meshes.__getitem__()
    • Meshes.isempty()
    • Meshes.verts_list()
    • Meshes.faces_list()
    • Meshes.verts_packed()
    • Meshes.verts_packed_to_mesh_idx()
    • Meshes.mesh_to_verts_packed_first_idx()
    • Meshes.num_verts_per_mesh()
    • Meshes.faces_packed()
    • Meshes.faces_packed_to_mesh_idx()
    • Meshes.mesh_to_faces_packed_first_idx()
    • Meshes.verts_padded()
    • Meshes.faces_padded()
    • Meshes.num_faces_per_mesh()
    • Meshes.edges_packed()
    • Meshes.edges_packed_to_mesh_idx()
    • Meshes.mesh_to_edges_packed_first_idx()
    • Meshes.faces_packed_to_edges_packed()
    • Meshes.num_edges_per_mesh()
    • Meshes.verts_padded_to_packed_idx()
    • Meshes.has_verts_normals()
    • Meshes.verts_normals_packed()
    • Meshes.verts_normals_list()
    • Meshes.verts_normals_padded()
    • Meshes.faces_normals_packed()
    • Meshes.faces_normals_list()
    • Meshes.faces_normals_padded()
    • Meshes.faces_areas_packed()
    • Meshes.laplacian_packed()
    • Meshes.clone()
    • Meshes.detach()
    • Meshes.to()
    • Meshes.cpu()
    • Meshes.cuda()
    • Meshes.get_mesh_verts_faces()
    • Meshes.split()
    • Meshes.offset_verts_()
    • Meshes.offset_verts()
    • Meshes.scale_verts_()
    • Meshes.scale_verts()
    • Meshes.update_padded()
    • Meshes.get_bounding_boxes()
    • Meshes.extend()
    • Meshes.sample_textures()
    • Meshes.submeshes()
  • join_pointclouds_as_batch()
  • join_pointclouds_as_scene()
  • Pointclouds
    • Pointclouds.__init__()
    • Pointclouds.__getitem__()
    • Pointclouds.isempty()
    • Pointclouds.points_list()
    • Pointclouds.normals_list()
    • Pointclouds.features_list()
    • Pointclouds.points_packed()
    • Pointclouds.normals_packed()
    • Pointclouds.features_packed()
    • Pointclouds.packed_to_cloud_idx()
    • Pointclouds.cloud_to_packed_first_idx()
    • Pointclouds.num_points_per_cloud()
    • Pointclouds.points_padded()
    • Pointclouds.normals_padded()
    • Pointclouds.features_padded()
    • Pointclouds.padded_to_packed_idx()
    • Pointclouds.clone()
    • Pointclouds.detach()
    • Pointclouds.to()
    • Pointclouds.cpu()
    • Pointclouds.cuda()
    • Pointclouds.get_cloud()
    • Pointclouds.split()
    • Pointclouds.offset_()
    • Pointclouds.offset()
    • Pointclouds.subsample()
    • Pointclouds.scale_()
    • Pointclouds.scale()
    • Pointclouds.get_bounding_boxes()
    • Pointclouds.estimate_normals()
    • Pointclouds.extend()
    • Pointclouds.update_padded()
    • Pointclouds.inside_box()
  • list_to_packed()
  • list_to_padded()
  • packed_to_list()
  • padded_to_list()
  • Volumes
    • Volumes.__init__()
    • Volumes.__getitem__()
    • Volumes.features()
    • Volumes.densities()
    • Volumes.densities_list()
    • Volumes.features_list()
    • Volumes.update_padded()
    • Volumes.clone()
    • Volumes.to()
    • Volumes.cpu()
    • Volumes.cuda()
    • Volumes.get_grid_sizes()
    • Volumes.get_local_to_world_coords_transform()
    • Volumes.get_world_to_local_coords_transform()
    • Volumes.world_to_local_coords()
    • Volumes.local_to_world_coords()
    • Volumes.get_coord_grid()
• pytorch3d.io
  • load_obj()
  • load_objs_as_meshes()
  • save_obj()
  • IO
    • IO.register_default_formats()
    • IO.register_meshes_format()
    • IO.register_pointcloud_format()
    • IO.load_mesh()
    • IO.save_mesh()
    • IO.load_pointcloud()
    • IO.save_pointcloud()
  • load_ply()
  • save_ply()
• pytorch3d.loss
  • chamfer_distance()
  • mesh_edge_loss()
  • mesh_laplacian_smoothing()
  • mesh_normal_consistency()
  • point_mesh_edge_distance()
  • point_mesh_face_distance()
• pytorch3d.ops
  • ball_query()
  • corresponding_cameras_alignment()
  • cubify()
  • GraphConv
    • GraphConv.__init__()
    • GraphConv.forward()
  • interpolate_face_attributes()
  • box3d_overlap()
  • knn_gather()
  • knn_points()
  • cot_laplacian()
  • laplacian()
  • norm_laplacian()
  • mesh_face_areas_normals()
  • taubin_smoothing()
  • packed_to_padded()
  • padded_to_packed()
  • efficient_pnp()
  • corresponding_points_alignment()
  • iterative_closest_point()
  • estimate_pointcloud_local_coord_frames()
  • estimate_pointcloud_normals()
  • add_pointclouds_to_volumes()
  • add_points_features_to_volume_densities_features()
  • sample_farthest_points()
  • sample_points_from_meshes()
  • SubdivideMeshes
    • SubdivideMeshes.__init__()
    • SubdivideMeshes.subdivide_faces()
    • SubdivideMeshes.forward()
    • SubdivideMeshes.subdivide_homogeneous()
    • SubdivideMeshes.subdivide_heterogenerous()
  • convert_pointclouds_to_tensor()
  • eyes()
  • get_point_covariances()
  • is_pointclouds()
  • wmean()
  • vert_align()
• pytorch3d.renderer
  • pytorch3d.renderer.blending
    • BlendParams
      • BlendParams.sigma
      • BlendParams.gamma
      • BlendParams.background_color
    • hard_rgb_blend()
    • sigmoid_alpha_blend()
    • softmax_rgb_blend()
  • pytorch3d.renderer.camera_conversions
  • pytorch3d.renderer.camera_utils
    • camera_to_eye_at_up()
    • rotate_on_spot()
    • join_cameras_as_batch()
  • pytorch3d.renderer.cameras
    • CamerasBase
      • CamerasBase.get_projection_transform()
      • CamerasBase.unproject_points()
      • CamerasBase.get_camera_center()
      • CamerasBase.get_world_to_view_transform()
      • CamerasBase.get_full_projection_transform()
      • CamerasBase.transform_points()
      • CamerasBase.get_ndc_camera_transform()
      • CamerasBase.transform_points_ndc()
      • CamerasBase.transform_points_screen()
      • CamerasBase.clone()
      • CamerasBase.is_perspective()
      • CamerasBase.in_ndc()
      • CamerasBase.get_znear()
      • CamerasBase.get_image_size()
      • CamerasBase.__getitem__()
    • OpenGLPerspectiveCameras()
    • FoVPerspectiveCameras
      • FoVPerspectiveCameras.__init__()
      • FoVPerspectiveCameras.compute_projection_matrix()
      • FoVPerspectiveCameras.get_projection_transform()
      • FoVPerspectiveCameras.unproject_points()
      • FoVPerspectiveCameras.is_perspective()
      • FoVPerspectiveCameras.in_ndc()
    • OpenGLOrthographicCameras()
    • FoVOrthographicCameras
      • FoVOrthographicCameras.__init__()
      • FoVOrthographicCameras.compute_projection_matrix()
      • FoVOrthographicCameras.get_projection_transform()
      • FoVOrthographicCameras.unproject_points()
      • FoVOrthographicCameras.is_perspective()
      • FoVOrthographicCameras.in_ndc()
    • SfMPerspectiveCameras()
    • PerspectiveCameras
      • PerspectiveCameras.__init__()
      • PerspectiveCameras.get_projection_transform()
      • PerspectiveCameras.unproject_points()
      • PerspectiveCameras.get_principal_point()
      • PerspectiveCameras.get_ndc_camera_transform()
      • PerspectiveCameras.is_perspective()
      • PerspectiveCameras.in_ndc()
    • SfMOrthographicCameras()
    • OrthographicCameras
      • OrthographicCameras.__init__()
      • OrthographicCameras.get_projection_transform()
      • OrthographicCameras.unproject_points()
      • OrthographicCameras.get_principal_point()
      • OrthographicCameras.get_ndc_camera_transform()
      • OrthographicCameras.is_perspective()
      • OrthographicCameras.in_ndc()
    • get_world_to_view_transform()
    • camera_position_from_spherical_angles()
    • look_at_rotation()
    • look_at_view_transform()
    • get_ndc_to_screen_transform()
    • get_screen_to_ndc_transform()
    • try_get_projection_transform()
  • pytorch3d.renderer.compositing
    • alpha_composite()
    • norm_weighted_sum()
    • weighted_sum()
  • pytorch3d.renderer.fisheyecameras
    • FishEyeCameras
      • FishEyeCameras.__init__()
      • FishEyeCameras.check_input()
      • FishEyeCameras.transform_points()
      • FishEyeCameras.unproject_points()
      • FishEyeCameras.in_ndc()
      • FishEyeCameras.is_perspective()
  • pytorch3d.renderer.lighting
    • diffuse()
    • specular()
    • DirectionalLights
      • DirectionalLights.__init__()
      • DirectionalLights.clone()
      • DirectionalLights.diffuse()
      • DirectionalLights.specular()
    • PointLights
      • PointLights.__init__()
      • PointLights.clone()
      • PointLights.reshape_location()
      • PointLights.diffuse()
      • PointLights.specular()
    • AmbientLights
      • AmbientLights.__init__()
      • AmbientLights.clone()
      • AmbientLights.diffuse()
      • AmbientLights.specular()
  • pytorch3d.renderer.materials
    • Materials
      • Materials.__init__()
      • Materials.clone()
  • pytorch3d.renderer.splatter_blend
    • SplatterBlender
      • SplatterBlender.__init__()
      • SplatterBlender.to()
      • SplatterBlender.forward()
  • pytorch3d.renderer.utils
    • TensorAccessor
      • TensorAccessor.__init__()
      • TensorAccessor.__setattr__()
      • TensorAccessor.__getattr__()
    • TensorProperties
      • TensorProperties.__init__()
      • TensorProperties.isempty()
      • TensorProperties.__getitem__()
      • TensorProperties.to()
      • TensorProperties.cpu()
      • TensorProperties.cuda()
      • TensorProperties.clone()
      • TensorProperties.gather_props()
    • format_tensor()
    • convert_to_tensors_and_broadcast()
    • ndc_grid_sample()
    • ndc_to_grid_sample_coords()
    • parse_image_size()
  • pytorch3d.renderer.implicit
    • pytorch3d.renderer.implicit.harmonic_embedding
      • HarmonicEmbedding
        • HarmonicEmbedding.__init__()
        • HarmonicEmbedding.forward()
        • HarmonicEmbedding.get_output_dim_static()
        • HarmonicEmbedding.get_output_dim()
    • pytorch3d.renderer.implicit.raymarching
      • EmissionAbsorptionRaymarcher
        • EmissionAbsorptionRaymarcher.__init__()
        • EmissionAbsorptionRaymarcher.forward()
      • AbsorptionOnlyRaymarcher
        • AbsorptionOnlyRaymarcher.forward()
    • pytorch3d.renderer.implicit.raysampling
      • MultinomialRaysampler
        • MultinomialRaysampler.min_x
        • MultinomialRaysampler.max_x
        • MultinomialRaysampler.min_y
        • MultinomialRaysampler.max_y
        • MultinomialRaysampler.__init__()
        • MultinomialRaysampler.forward()
      • NDCMultinomialRaysampler
      • MonteCarloRaysampler
        • MonteCarloRaysampler.__init__()
        • MonteCarloRaysampler.forward()
      • GridRaysampler()
      • NDCGridRaysampler()
    • pytorch3d.renderer.implicit.renderer
      • ImplicitRenderer
        • ImplicitRenderer.__init__()
        • ImplicitRenderer.forward()
      • VolumeRenderer
        • VolumeRenderer.__init__()
        • VolumeRenderer.forward()
      • VolumeSampler
        • VolumeSampler.__init__()
        • VolumeSampler.forward()
    • pytorch3d.renderer.implicit.sample_pdf
      • sample_pdf()
      • sample_pdf_python()
    • pytorch3d.renderer.implicit.utils
      • RayBundle
        • RayBundle.origins
        • RayBundle.directions
        • RayBundle.lengths
        • RayBundle.xys
      • HeterogeneousRayBundle
        • HeterogeneousRayBundle.origins
        • HeterogeneousRayBundle.directions
        • HeterogeneousRayBundle.lengths
        • HeterogeneousRayBundle.xys
        • HeterogeneousRayBundle.camera_ids
        • HeterogeneousRayBundle.camera_counts
      • ray_bundle_to_ray_points()
      • ray_bundle_variables_to_ray_points()
  • pytorch3d.renderer.mesh
    • pytorch3d.renderer.mesh.clip
      • ClippedFaces
        • ClippedFaces.face_verts
        • ClippedFaces.mesh_to_face_first_idx
        • ClippedFaces.num_faces_per_mesh
        • ClippedFaces.faces_clipped_to_unclipped_idx
        • ClippedFaces.barycentric_conversion
        • ClippedFaces.faces_clipped_to_conversion_idx
        • ClippedFaces.clipped_faces_neighbor_idx
      • ClipFrustum
        • ClipFrustum.left
        • ClipFrustum.right
        • ClipFrustum.top
        • ClipFrustum.bottom
        • ClipFrustum.znear
        • ClipFrustum.zfar
        • ClipFrustum.perspective_correct
        • ClipFrustum.cull
        • ClipFrustum.z_clip_value
      • clip_faces()
      • convert_clipped_rasterization_to_original_faces()
    • pytorch3d.renderer.mesh.rasterize_meshes
      • rasterize_meshes()
      • non_square_ndc_range()
      • pix_to_non_square_ndc()
      • rasterize_meshes_python()
      • edge_function()
      • barycentric_coordinates_clip()
      • barycentric_coordinates()
      • point_line_distance()
      • point_triangle_distance()
    • pytorch3d.renderer.mesh.rasterizer
      • Fragments
        • Fragments.pix_to_face
        • Fragments.zbuf
        • Fragments.bary_coords
        • Fragments.dists
        • Fragments.detach()
      • RasterizationSettings
        • RasterizationSettings.image_size
        • RasterizationSettings.blur_radius
        • RasterizationSettings.faces_per_pixel
        • RasterizationSettings.bin_size
        • RasterizationSettings.max_faces_opengl
        • RasterizationSettings.max_faces_per_bin
        • RasterizationSettings.perspective_correct
        • RasterizationSettings.clip_barycentric_coords
        • RasterizationSettings.cull_backfaces
        • RasterizationSettings.z_clip_value
        • RasterizationSettings.cull_to_frustum
      • MeshRasterizer
        • MeshRasterizer.__init__()
        • MeshRasterizer.to()
        • MeshRasterizer.transform()
        • MeshRasterizer.forward()
    • pytorch3d.renderer.mesh.renderer
      • MeshRenderer
        • MeshRenderer.to()
        • MeshRenderer.forward()
      • MeshRendererWithFragments
        • MeshRendererWithFragments.to()
        • MeshRendererWithFragments.forward()
    • pytorch3d.renderer.mesh.shader
      • ShaderBase
        • ShaderBase.to()
      • HardPhongShader
        • HardPhongShader.forward()
      • SoftPhongShader
        • SoftPhongShader.forward()
      • HardGouraudShader
        • HardGouraudShader.forward()
      • SoftGouraudShader
        • SoftGouraudShader.forward()
      • TexturedSoftPhongShader()
      • HardFlatShader
        • HardFlatShader.forward()
      • SoftSilhouetteShader
        • SoftSilhouetteShader.forward()
      • SplatterPhongShader
        • SplatterPhongShader.to()
        • SplatterPhongShader.forward()
        • SplatterPhongShader.check_blend_params()
      • HardDepthShader
        • HardDepthShader.forward()
      • SoftDepthShader
        • SoftDepthShader.forward()
    • pytorch3d.renderer.mesh.shading
      • phong_shading()
      • gouraud_shading()
      • flat_shading()
    • pytorch3d.renderer.mesh.textures
      • TexturesBase
        • TexturesBase.isempty()
        • TexturesBase.to()
        • TexturesBase.sample_textures()
        • TexturesBase.submeshes()
        • TexturesBase.faces_verts_textures_packed()
        • TexturesBase.clone()
        • TexturesBase.detach()
        • TexturesBase.__getitem__()
      • Textures()
      • TexturesAtlas
        • TexturesAtlas.__init__()
        • TexturesAtlas.clone()
        • TexturesAtlas.detach()
        • TexturesAtlas.atlas_padded()
        • TexturesAtlas.atlas_list()
        • TexturesAtlas.atlas_packed()
        • TexturesAtlas.extend()
        • TexturesAtlas.sample_textures()
        • TexturesAtlas.faces_verts_textures_packed()
        • TexturesAtlas.join_batch()
        • TexturesAtlas.join_scene()
        • TexturesAtlas.check_shapes()
      • TexturesUV
        • TexturesUV.__init__()
        • TexturesUV.clone()
        • TexturesUV.detach()
        • TexturesUV.faces_uvs_padded()
        • TexturesUV.faces_uvs_list()
        • TexturesUV.verts_uvs_padded()
        • TexturesUV.verts_uvs_list()
        • TexturesUV.maps_padded()
        • TexturesUV.maps_list()
        • TexturesUV.extend()
        • TexturesUV.sample_textures()
        • TexturesUV.faces_verts_textures_packed()
        • TexturesUV.join_batch()
        • TexturesUV.join_scene()
        • TexturesUV.centers_for_image()
        • TexturesUV.check_shapes()
      • TexturesVertex
        • TexturesVertex.__init__()
        • TexturesVertex.clone()
        • TexturesVertex.detach()
        • TexturesVertex.verts_features_padded()
        • TexturesVertex.verts_features_list()
        • TexturesVertex.verts_features_packed()
        • TexturesVertex.extend()
        • TexturesVertex.sample_textures()
        • TexturesVertex.submeshes()
        • TexturesVertex.faces_verts_textures_packed()
        • TexturesVertex.join_batch()
        • TexturesVertex.join_scene()
        • TexturesVertex.check_shapes()
    • pytorch3d.renderer.mesh.utils
      • Rectangle
        • Rectangle.xsize
        • Rectangle.ysize
        • Rectangle.identifier
      • PackedRectangle
        • PackedRectangle.x
        • PackedRectangle.y
        • PackedRectangle.flipped
        • PackedRectangle.is_first
      • PackedRectangles
        • PackedRectangles.total_size
        • PackedRectangles.locations
      • pack_rectangles()
      • pack_unique_rectangles()
  • pytorch3d.renderer.opengl
    • pytorch3d.renderer.opengl.opengl_utils
    • pytorch3d.renderer.opengl.rasterizer_opengl
  • pytorch3d.renderer.points
    • pytorch3d.renderer.points.compositor
      • AlphaCompositor
        • AlphaCompositor.forward()
      • NormWeightedCompositor
        • NormWeightedCompositor.forward()
    • pytorch3d.renderer.points.rasterize_points
      • rasterize_points()
      • rasterize_points_python()
    • pytorch3d.renderer.points.rasterizer
      • PointFragments
        • PointFragments.idx
        • PointFragments.zbuf
        • PointFragments.dists
      • PointsRasterizationSettings
        • PointsRasterizationSettings.image_size
        • PointsRasterizationSettings.radius
        • PointsRasterizationSettings.points_per_pixel
        • PointsRasterizationSettings.bin_size
        • PointsRasterizationSettings.max_points_per_bin
      • PointsRasterizer
        • PointsRasterizer.__init__()
        • PointsRasterizer.transform()
        • PointsRasterizer.to()
        • PointsRasterizer.forward()
    • pytorch3d.renderer.points.renderer
      • PointsRenderer
        • PointsRenderer.to()
        • PointsRenderer.forward()
    • pytorch3d.renderer.points.pulsar
      • pytorch3d.renderer.points.pulsar.renderer
        • Renderer
          • Renderer.sphere_ids_from_result_info_nograd()
          • Renderer.depth_map_from_result_info_nograd()
          • Renderer.forward()
          • Renderer.extra_repr()
      • pytorch3d.renderer.points.pulsar.unified
        • PulsarPointsRenderer
          • PulsarPointsRenderer.__init__()
          • PulsarPointsRenderer.forward()
• pytorch3d.transforms
  • acos_linear_extrapolation()
  • axis_angle_to_matrix()
  • axis_angle_to_quaternion()
  • euler_angles_to_matrix()
  • matrix_to_axis_angle()
  • matrix_to_euler_angles()
  • matrix_to_quaternion()
  • matrix_to_rotation_6d()
  • quaternion_apply()
  • quaternion_invert()
  • quaternion_multiply()
  • quaternion_raw_multiply()
  • quaternion_to_axis_angle()
  • quaternion_to_matrix()
  • random_quaternions()
  • random_rotation()
  • random_rotations()
  • rotation_6d_to_matrix()
  • standardize_quaternion()
  • se3_exp_map()
  • se3_log_map()
  • so3_exp_map()
  • so3_exponential_map()
  • so3_log_map()
  • so3_relative_angle()
  • so3_rotation_angle()
  • Rotate
    • Rotate.__init__()
  • RotateAxisAngle
    • RotateAxisAngle.__init__()
  • Scale
    • Scale.__init__()
  • Transform3d
    • Transform3d.__init__()
    • Transform3d.__getitem__()
    • Transform3d.compose()
    • Transform3d.get_matrix()
    • Transform3d.get_se3_log()
    • Transform3d.inverse()
    • Transform3d.stack()
    • Transform3d.transform_points()
    • Transform3d.transform_normals()
    • Transform3d.translate()
    • Transform3d.scale()
    • Transform3d.rotate()
    • Transform3d.rotate_axis_angle()
    • Transform3d.clone()
    • Transform3d.to()
    • Transform3d.cpu()
    • Transform3d.cuda()
  • Translate
    • Translate.__init__()
• pytorch3d.utils
  • cameras_from_opencv_projection()
  • opencv_from_cameras_projection()
  • pulsar_from_cameras_projection()
  • pulsar_from_opencv_projection()
  • checkerboard()
  • ico_sphere()
  • torus()
• pytorch3d.datasets
  • BlenderCamera
    • BlenderCamera.__init__()
    • BlenderCamera.get_projection_transform()
    • BlenderCamera.is_perspective()
    • BlenderCamera.in_ndc()
  • collate_batched_R2N2()
  • R2N2
    • R2N2.__init__()
    • R2N2.__getitem__()
    • R2N2.render()
  • render_cubified_voxels()
  • ShapeNetCore
    • ShapeNetCore.__init__()
    • ShapeNetCore.__getitem__()
  • collate_batched_meshes()
• pytorch3d.common
  • get_device()
  • make_device()
• pytorch3d.vis
  • pytorch3d.vis.plotly_vis
  • pytorch3d.vis.texture_vis
    • texturesuv_image_matplotlib()
    • texturesuv_image_PIL()
• pytorch3d.implicitron
  • pytorch3d.implicitron.models
    • pytorch3d.implicitron.models.base_model
      • ImplicitronRender
        • ImplicitronRender.depth_render
        • ImplicitronRender.image_render
        • ImplicitronRender.mask_render
        • ImplicitronRender.camera_distance
        • ImplicitronRender.clone()
      • ImplicitronModelBase
        • ImplicitronModelBase.log_vars
        • ImplicitronModelBase.forward()
    • pytorch3d.implicitron.models.generic_model
      • GenericModel
        • GenericModel.mask_images
        • GenericModel.mask_depths
        • GenericModel.render_image_width
        • GenericModel.render_image_height
        • GenericModel.mask_threshold
        • GenericModel.output_rasterized_mc
        • GenericModel.bg_color
        • GenericModel.num_passes
        • GenericModel.chunk_size_grid
        • GenericModel.render_features_dimensions
        • GenericModel.tqdm_trigger_threshold
        • GenericModel.n_train_target_views
        • GenericModel.sampling_mode_training
        • GenericModel.sampling_mode_evaluation
        • GenericModel.global_encoder_class_type
        • GenericModel.global_encoder
        • GenericModel.raysampler_class_type
        • GenericModel.raysampler
        • GenericModel.renderer_class_type
        • GenericModel.renderer
        • GenericModel.image_feature_extractor
        • GenericModel.image_feature_extractor_class_type
        • GenericModel.view_pooler_enabled
        • GenericModel.view_pooler
        • GenericModel.implicit_function_class_type
        • GenericModel.implicit_function
        • GenericModel.view_metrics
        • GenericModel.view_metrics_class_type
        • GenericModel.regularization_metrics
        • GenericModel.regularization_metrics_class_type
        • GenericModel.loss_weights
        • GenericModel.log_vars
        • GenericModel.pre_expand()
        • GenericModel.forward()
        • GenericModel.visualize()
        • GenericModel.raysampler_tweak_args()
        • GenericModel.create_raysampler()
        • GenericModel.renderer_tweak_args()
        • GenericModel.create_renderer()
        • GenericModel.create_implicit_function()
        • GenericModel.implicit_function_tweak_args()
    • pytorch3d.implicitron.models.metrics
      • RegularizationMetricsBase
        • RegularizationMetricsBase.forward()
      • ViewMetricsBase
        • ViewMetricsBase.forward()
      • RegularizationMetrics
        • RegularizationMetrics.forward()
      • ViewMetrics
        • ViewMetrics.forward()
    • pytorch3d.implicitron.models.model_dbir
      • ModelDBIR
        • ModelDBIR.render_image_width
        • ModelDBIR.render_image_height
        • ModelDBIR.bg_color
        • ModelDBIR.max_points
        • ModelDBIR.forward()
    • pytorch3d.implicitron.models.feature_extractor
      • pytorch3d.implicitron.models.feature_extractor.feature_extractor
        • FeatureExtractorBase
          • FeatureExtractorBase.get_feat_dims()
          • FeatureExtractorBase.forward()
      • pytorch3d.implicitron.models.feature_extractor.resnet_feature_extractor
        • ResNetFeatureExtractor
          • ResNetFeatureExtractor.name
          • ResNetFeatureExtractor.pretrained
          • ResNetFeatureExtractor.stages
          • ResNetFeatureExtractor.normalize_image
          • ResNetFeatureExtractor.image_rescale
          • ResNetFeatureExtractor.first_max_pool
          • ResNetFeatureExtractor.proj_dim
          • ResNetFeatureExtractor.l2_norm
          • ResNetFeatureExtractor.add_masks
          • ResNetFeatureExtractor.add_images
          • ResNetFeatureExtractor.global_average_pool
          • ResNetFeatureExtractor.feature_rescale
          • ResNetFeatureExtractor.get_feat_dims()
          • ResNetFeatureExtractor.forward()
    • pytorch3d.implicitron.models.global_encoder
      • pytorch3d.implicitron.models.global_encoder.autodecoder
        • Autodecoder
          • Autodecoder.encoding_dim
          • Autodecoder.n_instances
          • Autodecoder.init_scale
          • Autodecoder.ignore_input
          • Autodecoder.calculate_squared_encoding_norm()
          • Autodecoder.get_encoding_dim()
          • Autodecoder.forward()
      • pytorch3d.implicitron.models.global_encoder.global_encoder
        • GlobalEncoderBase
          • GlobalEncoderBase.get_encoding_dim()
          • GlobalEncoderBase.calculate_squared_encoding_norm()
          • GlobalEncoderBase.forward()
        • SequenceAutodecoder
          • SequenceAutodecoder.autodecoder
          • SequenceAutodecoder.get_encoding_dim()
          • SequenceAutodecoder.forward()
          • SequenceAutodecoder.calculate_squared_encoding_norm()
        • HarmonicTimeEncoder
          • HarmonicTimeEncoder.n_harmonic_functions
          • HarmonicTimeEncoder.append_input
          • HarmonicTimeEncoder.time_divisor
          • HarmonicTimeEncoder.get_encoding_dim()
          • HarmonicTimeEncoder.forward()
          • HarmonicTimeEncoder.calculate_squared_encoding_norm()
    • pytorch3d.implicitron.models.implicit_function
      • pytorch3d.implicitron.models.implicit_function.base
        • ImplicitFunctionBase
          • ImplicitFunctionBase.forward()
          • ImplicitFunctionBase.allows_multiple_passes()
          • ImplicitFunctionBase.requires_pooling_without_aggregation()
          • ImplicitFunctionBase.on_bind_args()
      • pytorch3d.implicitron.models.implicit_function.decoding_functions
        • DecoderActivation
          • DecoderActivation.RELU
          • DecoderActivation.SOFTPLUS
          • DecoderActivation.SIGMOID
          • DecoderActivation.IDENTITY
        • DecoderFunctionBase
          • DecoderFunctionBase.forward()
        • ElementwiseDecoder
          • ElementwiseDecoder.scale
          • ElementwiseDecoder.shift
          • ElementwiseDecoder.operation
          • ElementwiseDecoder.forward()
        • MLPWithInputSkips
          • MLPWithInputSkips.n_layers
          • MLPWithInputSkips.input_dim
          • MLPWithInputSkips.output_dim
          • MLPWithInputSkips.skip_dim
          • MLPWithInputSkips.hidden_dim
          • MLPWithInputSkips.input_skips
          • MLPWithInputSkips.skip_affine_trans
          • MLPWithInputSkips.last_layer_bias_init
          • MLPWithInputSkips.last_activation
          • MLPWithInputSkips.use_xavier_init
          • MLPWithInputSkips.forward()
        • MLPDecoder
          • MLPDecoder.input_dim
          • MLPDecoder.param_groups
          • MLPDecoder.network
          • MLPDecoder.forward()
          • MLPDecoder.network_tweak_args()
          • MLPDecoder.create_network_impl()
        • TransformerWithInputSkips
          • TransformerWithInputSkips.__init__()
          • TransformerWithInputSkips.forward()
        • TransformerEncoderLayer
          • TransformerEncoderLayer.forward()
      • pytorch3d.implicitron.models.implicit_function.idr_feature_field
        • IdrFeatureField
          • IdrFeatureField.feature_vector_size
          • IdrFeatureField.d_in
          • IdrFeatureField.d_out
          • IdrFeatureField.dims
          • IdrFeatureField.geometric_init
          • IdrFeatureField.bias
          • IdrFeatureField.skip_in
          • IdrFeatureField.weight_norm
          • IdrFeatureField.n_harmonic_functions_xyz
          • IdrFeatureField.pooled_feature_dim
          • IdrFeatureField.encoding_dim
          • IdrFeatureField.forward()
      • pytorch3d.implicitron.models.implicit_function.neural_radiance_field
        • NeuralRadianceFieldBase
          • NeuralRadianceFieldBase.n_harmonic_functions_xyz
          • NeuralRadianceFieldBase.n_harmonic_functions_dir
          • NeuralRadianceFieldBase.n_hidden_neurons_dir
          • NeuralRadianceFieldBase.latent_dim
          • NeuralRadianceFieldBase.input_xyz
          • NeuralRadianceFieldBase.xyz_ray_dir_in_camera_coords
          • NeuralRadianceFieldBase.color_dim
          • NeuralRadianceFieldBase.use_integrated_positional_encoding
          • NeuralRadianceFieldBase.get_xyz_embedding_dim()
          • NeuralRadianceFieldBase.allows_multiple_passes()
          • NeuralRadianceFieldBase.forward()
        • NeuralRadianceFieldImplicitFunction
          • NeuralRadianceFieldImplicitFunction.transformer_dim_down_factor
          • NeuralRadianceFieldImplicitFunction.n_hidden_neurons_xyz
          • NeuralRadianceFieldImplicitFunction.n_layers_xyz
          • NeuralRadianceFieldImplicitFunction.append_xyz
        • NeRFormerImplicitFunction
          • NeRFormerImplicitFunction.transformer_dim_down_factor
          • NeRFormerImplicitFunction.n_hidden_neurons_xyz
          • NeRFormerImplicitFunction.n_layers_xyz
          • NeRFormerImplicitFunction.append_xyz
          • NeRFormerImplicitFunction.requires_pooling_without_aggregation()
      • pytorch3d.implicitron.models.implicit_function.scene_representation_networks
        • SRNRaymarchFunction
          • SRNRaymarchFunction.n_harmonic_functions
          • SRNRaymarchFunction.n_hidden_units
          • SRNRaymarchFunction.n_layers
          • SRNRaymarchFunction.in_features
          • SRNRaymarchFunction.out_features
          • SRNRaymarchFunction.latent_dim
          • SRNRaymarchFunction.xyz_in_camera_coords
          • SRNRaymarchFunction.raymarch_function
          • SRNRaymarchFunction.forward()
        • SRNPixelGenerator
          • SRNPixelGenerator.n_harmonic_functions
          • SRNPixelGenerator.n_hidden_units
          • SRNPixelGenerator.n_hidden_units_color
          • SRNPixelGenerator.n_layers
          • SRNPixelGenerator.in_features
          • SRNPixelGenerator.out_features
          • SRNPixelGenerator.ray_dir_in_camera_coords
          • SRNPixelGenerator.forward()
        • SRNRaymarchHyperNet
          • SRNRaymarchHyperNet.n_harmonic_functions
          • SRNRaymarchHyperNet.n_hidden_units
          • SRNRaymarchHyperNet.n_layers
          • SRNRaymarchHyperNet.n_hidden_units_hypernet
          • SRNRaymarchHyperNet.n_layers_hypernet
          • SRNRaymarchHyperNet.in_features
          • SRNRaymarchHyperNet.out_features
          • SRNRaymarchHyperNet.latent_dim_hypernet
          • SRNRaymarchHyperNet.latent_dim
          • SRNRaymarchHyperNet.xyz_in_camera_coords
          • SRNRaymarchHyperNet.forward()
        • SRNImplicitFunction
          • SRNImplicitFunction.latent_dim
          • SRNImplicitFunction.raymarch_function
          • SRNImplicitFunction.pixel_generator
          • SRNImplicitFunction.create_raymarch_function()
          • SRNImplicitFunction.raymarch_function_tweak_args()
          • SRNImplicitFunction.forward()
        • SRNHyperNetImplicitFunction
          • SRNHyperNetImplicitFunction.latent_dim_hypernet
          • SRNHyperNetImplicitFunction.latent_dim
          • SRNHyperNetImplicitFunction.hypernet
          • SRNHyperNetImplicitFunction.pixel_generator
          • SRNHyperNetImplicitFunction.create_hypernet()
          • SRNHyperNetImplicitFunction.hypernet_tweak_args()
          • SRNHyperNetImplicitFunction.forward()
          • SRNHyperNetImplicitFunction.on_bind_args()
      • pytorch3d.implicitron.models.implicit_function.utils
        • broadcast_global_code()
        • create_embeddings_for_implicit_function()
        • interpolate_line()
        • interpolate_plane()
        • interpolate_volume()
        • get_rays_points_world()
      • pytorch3d.implicitron.models.implicit_function.voxel_grid
        • VoxelGridValuesBase
        • VoxelGridBase
          • VoxelGridBase.align_corners
          • VoxelGridBase.padding
          • VoxelGridBase.mode
          • VoxelGridBase.n_features
          • VoxelGridBase.resolution_changes
          • VoxelGridBase.evaluate_world()
          • VoxelGridBase.evaluate_local()
          • VoxelGridBase.get_shapes()
          • VoxelGridBase.get_resolution()
          • VoxelGridBase.get_output_dim()
          • VoxelGridBase.change_resolution()
          • VoxelGridBase.get_resolution_change_epochs()
          • VoxelGridBase.get_align_corners()
          • VoxelGridBase.crop_world()
          • VoxelGridBase.crop_local()
        • FullResolutionVoxelGridValues
          • FullResolutionVoxelGridValues.voxel_grid
        • FullResolutionVoxelGrid
          • FullResolutionVoxelGrid.values_type
          • FullResolutionVoxelGrid.evaluate_local()
          • FullResolutionVoxelGrid.get_shapes()
          • FullResolutionVoxelGrid.crop_local()
        • CPFactorizedVoxelGridValues
          • CPFactorizedVoxelGridValues.vector_components_x
          • CPFactorizedVoxelGridValues.vector_components_y
          • CPFactorizedVoxelGridValues.vector_components_z
          • CPFactorizedVoxelGridValues.basis_matrix
        • CPFactorizedVoxelGrid
          • CPFactorizedVoxelGrid.values_type
          • CPFactorizedVoxelGrid.n_components
          • CPFactorizedVoxelGrid.basis_matrix
          • CPFactorizedVoxelGrid.evaluate_local()
          • CPFactorizedVoxelGrid.get_shapes()
          • CPFactorizedVoxelGrid.crop_local()
        • VMFactorizedVoxelGridValues
          • VMFactorizedVoxelGridValues.vector_components_x
          • VMFactorizedVoxelGridValues.vector_components_y
          • VMFactorizedVoxelGridValues.vector_components_z
          • VMFactorizedVoxelGridValues.matrix_components_xy
          • VMFactorizedVoxelGridValues.matrix_components_yz
          • VMFactorizedVoxelGridValues.matrix_components_xz
          • VMFactorizedVoxelGridValues.basis_matrix
        • VMFactorizedVoxelGrid
          • VMFactorizedVoxelGrid.values_type
          • VMFactorizedVoxelGrid.n_components
          • VMFactorizedVoxelGrid.distribution_of_components
          • VMFactorizedVoxelGrid.basis_matrix
          • VMFactorizedVoxelGrid.evaluate_local()
          • VMFactorizedVoxelGrid.get_shapes()
          • VMFactorizedVoxelGrid.crop_local()
        • VoxelGridModule
          • VoxelGridModule.voxel_grid_class_type
          • VoxelGridModule.voxel_grid
          • VoxelGridModule.extents
          • VoxelGridModule.translation
          • VoxelGridModule.init_std
          • VoxelGridModule.init_mean
          • VoxelGridModule.hold_voxel_grid_as_parameters
          • VoxelGridModule.param_groups
          • VoxelGridModule.forward()
          • VoxelGridModule.set_voxel_grid_parameters()
          • VoxelGridModule.get_output_dim()
          • VoxelGridModule.subscribe_to_epochs()
          • VoxelGridModule.get_device()
          • VoxelGridModule.crop_self()
          • VoxelGridModule.get_grid_points()
      • pytorch3d.implicitron.models.implicit_function.voxel_grid_implicit_function
        • VoxelGridImplicitFunction
          • VoxelGridImplicitFunction.voxel_grid_density
          • VoxelGridImplicitFunction.voxel_grid_color
          • VoxelGridImplicitFunction.harmonic_embedder_xyz_density_args
          • VoxelGridImplicitFunction.harmonic_embedder_xyz_color_args
          • VoxelGridImplicitFunction.harmonic_embedder_dir_color_args
          • VoxelGridImplicitFunction.decoder_density_class_type
          • VoxelGridImplicitFunction.decoder_density
          • VoxelGridImplicitFunction.decoder_color_class_type
          • VoxelGridImplicitFunction.decoder_color
          • VoxelGridImplicitFunction.use_multiple_streams
          • VoxelGridImplicitFunction.xyz_ray_dir_in_camera_coords
          • VoxelGridImplicitFunction.scaffold_calculating_epochs
          • VoxelGridImplicitFunction.scaffold_resolution
          • VoxelGridImplicitFunction.scaffold_empty_space_threshold
          • VoxelGridImplicitFunction.scaffold_occupancy_chunk_size
          • VoxelGridImplicitFunction.scaffold_max_pool_kernel_size
          • VoxelGridImplicitFunction.scaffold_filter_points
          • VoxelGridImplicitFunction.volume_cropping_epochs
          • VoxelGridImplicitFunction.forward()
          • VoxelGridImplicitFunction.allows_multiple_passes()
          • VoxelGridImplicitFunction.subscribe_to_epochs()
          • VoxelGridImplicitFunction.decoder_density_tweak_args()
          • VoxelGridImplicitFunction.create_decoder_density_impl()
          • VoxelGridImplicitFunction.decoder_color_tweak_args()
          • VoxelGridImplicitFunction.create_decoder_color_impl()
    • pytorch3d.implicitron.models.renderer
      • pytorch3d.implicitron.models.renderer.base
        • EvaluationMode
          • EvaluationMode.TRAINING
          • EvaluationMode.EVALUATION
        • RenderSamplingMode
          • RenderSamplingMode.MASK_SAMPLE
          • RenderSamplingMode.FULL_GRID
        • ImplicitronRayBundle
          • ImplicitronRayBundle.origins
          • ImplicitronRayBundle.directions
          • ImplicitronRayBundle.lengths
          • ImplicitronRayBundle.xys
          • ImplicitronRayBundle.camera_ids
          • ImplicitronRayBundle.camera_counts
          • ImplicitronRayBundle.bins
          • ImplicitronRayBundle.pixel_radii_2d
          • ImplicitronRayBundle.lengths
          • ImplicitronRayBundle.is_packed()
          • ImplicitronRayBundle.get_padded_xys()
        • RendererOutput
          • RendererOutput.features
          • RendererOutput.depths
          • RendererOutput.masks
          • RendererOutput.prev_stage
          • RendererOutput.normals
          • RendererOutput.points
          • RendererOutput.weights
          • RendererOutput.aux
        • ImplicitFunctionWrapper
          • ImplicitFunctionWrapper.bind_args()
          • ImplicitFunctionWrapper.unbind_args()
          • ImplicitFunctionWrapper.forward()
        • BaseRenderer
          • BaseRenderer.requires_object_mask()
          • BaseRenderer.forward()
        • compute_3d_diagonal_covariance_gaussian()
        • approximate_conical_frustum_as_gaussians()
        • conical_frustum_to_gaussian()
      • pytorch3d.implicitron.models.renderer.lstm_renderer
        • LSTMRenderer
          • LSTMRenderer.num_raymarch_steps
          • LSTMRenderer.init_depth
          • LSTMRenderer.init_depth_noise_std
          • LSTMRenderer.hidden_size
          • LSTMRenderer.n_feature_channels
          • LSTMRenderer.bg_color
          • LSTMRenderer.verbose
          • LSTMRenderer.forward()
      • pytorch3d.implicitron.models.renderer.multipass_ea
        • MultiPassEmissionAbsorptionRenderer
          • MultiPassEmissionAbsorptionRenderer.raymarcher_class_type
          • MultiPassEmissionAbsorptionRenderer.raymarcher
          • MultiPassEmissionAbsorptionRenderer.n_pts_per_ray_fine_training
          • MultiPassEmissionAbsorptionRenderer.n_pts_per_ray_fine_evaluation
          • MultiPassEmissionAbsorptionRenderer.stratified_sampling_coarse_training
          • MultiPassEmissionAbsorptionRenderer.stratified_sampling_coarse_evaluation
          • MultiPassEmissionAbsorptionRenderer.append_coarse_samples_to_fine
          • MultiPassEmissionAbsorptionRenderer.density_noise_std_train
          • MultiPassEmissionAbsorptionRenderer.return_weights
          • MultiPassEmissionAbsorptionRenderer.blurpool_weights
          • MultiPassEmissionAbsorptionRenderer.sample_pdf_eps
          • MultiPassEmissionAbsorptionRenderer.forward()
      • pytorch3d.implicitron.models.renderer.ray_point_refiner
        • RayPointRefiner
          • RayPointRefiner.n_pts_per_ray
          • RayPointRefiner.random_sampling
          • RayPointRefiner.add_input_samples
          • RayPointRefiner.blurpool_weights
          • RayPointRefiner.sample_pdf_eps
          • RayPointRefiner.forward()
        • apply_blurpool_on_weights()
      • pytorch3d.implicitron.models.renderer.ray_sampler
        • RaySamplerBase
          • RaySamplerBase.forward()
        • AbstractMaskRaySampler
          • AbstractMaskRaySampler.image_width
          • AbstractMaskRaySampler.image_height
          • AbstractMaskRaySampler.sampling_mode_training
          • AbstractMaskRaySampler.sampling_mode_evaluation
          • AbstractMaskRaySampler.n_pts_per_ray_training
          • AbstractMaskRaySampler.n_pts_per_ray_evaluation
          • AbstractMaskRaySampler.n_rays_per_image_sampled_from_mask
          • AbstractMaskRaySampler.n_rays_total_training
          • AbstractMaskRaySampler.stratified_point_sampling_training
          • AbstractMaskRaySampler.stratified_point_sampling_evaluation
          • AbstractMaskRaySampler.cast_ray_bundle_as_cone
          • AbstractMaskRaySampler.forward()
        • AdaptiveRaySampler
          • AdaptiveRaySampler.scene_extent
          • AdaptiveRaySampler.scene_center
        • NearFarRaySampler
          • NearFarRaySampler.min_depth
          • NearFarRaySampler.max_depth
        • compute_radii()
      • pytorch3d.implicitron.models.renderer.ray_tracing
        • RayTracing
          • RayTracing.object_bounding_sphere
          • RayTracing.sdf_threshold
          • RayTracing.line_search_step
          • RayTracing.line_step_iters
          • RayTracing.sphere_tracing_iters
          • RayTracing.n_steps
          • RayTracing.n_secant_steps
          • RayTracing.forward()
          • RayTracing.sphere_tracing()
          • RayTracing.ray_sampler()
          • RayTracing.secant()
          • RayTracing.minimal_sdf_points()
      • pytorch3d.implicitron.models.renderer.raymarcher
        • RaymarcherBase
          • RaymarcherBase.forward()
        • AccumulativeRaymarcherBase
          • AccumulativeRaymarcherBase.surface_thickness
          • AccumulativeRaymarcherBase.bg_color
          • AccumulativeRaymarcherBase.replicate_last_interval
          • AccumulativeRaymarcherBase.background_opacity
          • AccumulativeRaymarcherBase.density_relu
          • AccumulativeRaymarcherBase.blend_output
          • AccumulativeRaymarcherBase.capping_function_type
          • AccumulativeRaymarcherBase.weight_function_type
          • AccumulativeRaymarcherBase.__post_init__()
          • AccumulativeRaymarcherBase.forward()
        • EmissionAbsorptionRaymarcher
          • EmissionAbsorptionRaymarcher.background_opacity
          • EmissionAbsorptionRaymarcher.capping_function_type
          • EmissionAbsorptionRaymarcher.weight_function_type
        • CumsumRaymarcher
          • CumsumRaymarcher.capping_function_type
          • CumsumRaymarcher.weight_function_type
      • pytorch3d.implicitron.models.renderer.rgb_net
        • RayNormalColoringNetwork
          • RayNormalColoringNetwork.forward()
      • pytorch3d.implicitron.models.renderer.sdf_renderer
        • SignedDistanceFunctionRenderer
          • SignedDistanceFunctionRenderer.render_features_dimensions
          • SignedDistanceFunctionRenderer.object_bounding_sphere
          • SignedDistanceFunctionRenderer.ray_tracer
          • SignedDistanceFunctionRenderer.ray_normal_coloring_network_args
          • SignedDistanceFunctionRenderer.bg_color
          • SignedDistanceFunctionRenderer.soft_mask_alpha
          • SignedDistanceFunctionRenderer.ray_tracer_tweak_args()
          • SignedDistanceFunctionRenderer.create_ray_tracer()
          • SignedDistanceFunctionRenderer.requires_object_mask()
          • SignedDistanceFunctionRenderer.forward()
    • pytorch3d.implicitron.models.view_pooler
      • pytorch3d.implicitron.models.view_pooler.feature_aggregator
        • ReductionFunction
          • ReductionFunction.AVG
          • ReductionFunction.MAX
          • ReductionFunction.STD
          • ReductionFunction.STD_AVG
        • FeatureAggregatorBase
          • FeatureAggregatorBase.exclude_target_view
          • FeatureAggregatorBase.exclude_target_view_mask_features
          • FeatureAggregatorBase.concatenate_output
          • FeatureAggregatorBase.forward()
          • FeatureAggregatorBase.get_aggregated_feature_dim()
          • FeatureAggregatorBase.has_aggregation()
        • IdentityFeatureAggregator
          • IdentityFeatureAggregator.get_aggregated_feature_dim()
          • IdentityFeatureAggregator.forward()
        • ReductionFeatureAggregator
          • ReductionFeatureAggregator.reduction_functions
          • ReductionFeatureAggregator.get_aggregated_feature_dim()
          • ReductionFeatureAggregator.forward()
        • AngleWeightedReductionFeatureAggregator
          • AngleWeightedReductionFeatureAggregator.reduction_functions
          • AngleWeightedReductionFeatureAggregator.weight_by_ray_angle_gamma
          • AngleWeightedReductionFeatureAggregator.min_ray_angle_weight
          • AngleWeightedReductionFeatureAggregator.get_aggregated_feature_dim()
          • AngleWeightedReductionFeatureAggregator.forward()
        • AngleWeightedIdentityFeatureAggregator
          • AngleWeightedIdentityFeatureAggregator.weight_by_ray_angle_gamma
          • AngleWeightedIdentityFeatureAggregator.min_ray_angle_weight
          • AngleWeightedIdentityFeatureAggregator.get_aggregated_feature_dim()
          • AngleWeightedIdentityFeatureAggregator.forward()
      • pytorch3d.implicitron.models.view_pooler.view_pooler
        • ViewPooler
          • ViewPooler.view_sampler
          • ViewPooler.feature_aggregator_class_type
          • ViewPooler.feature_aggregator
          • ViewPooler.get_aggregated_feature_dim()
          • ViewPooler.has_aggregation()
          • ViewPooler.forward()
      • pytorch3d.implicitron.models.view_pooler.view_sampler
        • ViewSampler
          • ViewSampler.masked_sampling
          • ViewSampler.sampling_mode
          • ViewSampler.forward()
        • project_points_and_sample()
        • handle_seq_id()
        • cameras_points_cartesian_product()
    • pytorch3d.implicitron.models.visualization
      • pytorch3d.implicitron.models.visualization.render_flyaround
  • pytorch3d.implicitron.dataset in general
    • DatasetBase
      • DatasetBase.get_frame_numbers_and_timestamps()
      • DatasetBase.join()
      • DatasetBase.get_eval_batches()
      • DatasetBase.sequence_names()
      • DatasetBase.category_to_sequence_names()
      • DatasetBase.sequence_frames_in_order()
      • DatasetBase.sequence_indices_in_order()
      • DatasetBase.frame_data_type
    • DatasetMap
      • DatasetMap.train
      • DatasetMap.val
      • DatasetMap.test
      • DatasetMap.__getitem__()
      • DatasetMap.iter_datasets()
      • DatasetMap.join()
    • DatasetMapProviderBase
      • DatasetMapProviderBase.get_dataset_map()
      • DatasetMapProviderBase.get_all_train_cameras()
    • PathManagerFactory
      • PathManagerFactory.silence_logs
      • PathManagerFactory.get()
    • DataLoaderMap
      • DataLoaderMap.train
      • DataLoaderMap.val
      • DataLoaderMap.test
      • DataLoaderMap.__getitem__()
    • DataLoaderMapProviderBase
      • DataLoaderMapProviderBase.get_data_loader_map()
    • SimpleDataLoaderMapProvider
      • SimpleDataLoaderMapProvider.batch_size
      • SimpleDataLoaderMapProvider.num_workers
      • SimpleDataLoaderMapProvider.dataset_length_train
      • SimpleDataLoaderMapProvider.dataset_length_val
      • SimpleDataLoaderMapProvider.dataset_length_test
      • SimpleDataLoaderMapProvider.get_data_loader_map()
    • DoublePoolBatchSampler
      • DoublePoolBatchSampler.__init__()
    • BatchConditioningType
      • BatchConditioningType.SAME
      • BatchConditioningType.TRAIN
      • BatchConditioningType.KNOWN
    • SequenceDataLoaderMapProvider
      • SequenceDataLoaderMapProvider.batch_size
      • SequenceDataLoaderMapProvider.num_workers
      • SequenceDataLoaderMapProvider.dataset_length_train
      • SequenceDataLoaderMapProvider.dataset_length_val
      • SequenceDataLoaderMapProvider.dataset_length_test
      • SequenceDataLoaderMapProvider.train_conditioning_type
      • SequenceDataLoaderMapProvider.val_conditioning_type
      • SequenceDataLoaderMapProvider.test_conditioning_type
      • SequenceDataLoaderMapProvider.images_per_seq_options
      • SequenceDataLoaderMapProvider.sample_consecutive_frames
      • SequenceDataLoaderMapProvider.consecutive_frames_max_gap
      • SequenceDataLoaderMapProvider.consecutive_frames_max_gap_seconds
      • SequenceDataLoaderMapProvider.get_data_loader_map()
    • DataSourceBase
      • DataSourceBase.get_datasets_and_dataloaders()
      • DataSourceBase.all_train_cameras
    • ImplicitronDataSource
      • ImplicitronDataSource.dataset_map_provider
      • ImplicitronDataSource.dataset_map_provider_class_type
      • ImplicitronDataSource.data_loader_map_provider
      • ImplicitronDataSource.data_loader_map_provider_class_type
      • ImplicitronDataSource.pre_expand()
      • ImplicitronDataSource.get_datasets_and_dataloaders()
      • ImplicitronDataSource.all_train_cameras
    • SceneBatchSampler
      • SceneBatchSampler.dataset
      • SceneBatchSampler.batch_size
      • SceneBatchSampler.num_batches
      • SceneBatchSampler.images_per_seq_options
      • SceneBatchSampler.sample_consecutive_frames
      • SceneBatchSampler.consecutive_frames_max_gap
      • SceneBatchSampler.consecutive_frames_max_gap_seconds
      • SceneBatchSampler.category_aware
      • SceneBatchSampler.seq_names
      • SceneBatchSampler.category_to_sequence_names
      • SceneBatchSampler.categories
  • pytorch3d.implicitron.dataset specific datasets
    • BlenderDatasetMapProvider
    • JsonIndexDatasetMapProvider
      • JsonIndexDatasetMapProvider.category
      • JsonIndexDatasetMapProvider.task_str
      • JsonIndexDatasetMapProvider.dataset_root
      • JsonIndexDatasetMapProvider.n_frames_per_sequence
      • JsonIndexDatasetMapProvider.test_on_train
      • JsonIndexDatasetMapProvider.restrict_sequence_name
      • JsonIndexDatasetMapProvider.test_restrict_sequence_id
      • JsonIndexDatasetMapProvider.assert_single_seq
      • JsonIndexDatasetMapProvider.only_test_set
      • JsonIndexDatasetMapProvider.dataset
      • JsonIndexDatasetMapProvider.dataset_class_type
      • JsonIndexDatasetMapProvider.path_manager_factory
      • JsonIndexDatasetMapProvider.path_manager_factory_class_type
      • JsonIndexDatasetMapProvider.dataset_tweak_args()
      • JsonIndexDatasetMapProvider.create_dataset()
      • JsonIndexDatasetMapProvider.get_dataset_map()
      • JsonIndexDatasetMapProvider.get_all_train_cameras()
    • JsonIndexDatasetMapProviderV2
      • JsonIndexDatasetMapProviderV2.category
      • JsonIndexDatasetMapProviderV2.subset_name
      • JsonIndexDatasetMapProviderV2.dataset_root
      • JsonIndexDatasetMapProviderV2.test_on_train
      • JsonIndexDatasetMapProviderV2.only_test_set
      • JsonIndexDatasetMapProviderV2.load_eval_batches
      • JsonIndexDatasetMapProviderV2.num_load_workers
      • JsonIndexDatasetMapProviderV2.n_known_frames_for_test
      • JsonIndexDatasetMapProviderV2.dataset_class_type
      • JsonIndexDatasetMapProviderV2.dataset
      • JsonIndexDatasetMapProviderV2.path_manager_factory
      • JsonIndexDatasetMapProviderV2.path_manager_factory_class_type
      • JsonIndexDatasetMapProviderV2.dataset_tweak_args()
      • JsonIndexDatasetMapProviderV2.create_dataset()
      • JsonIndexDatasetMapProviderV2.get_dataset_map()
      • JsonIndexDatasetMapProviderV2.get_category_to_subset_name_list()
      • JsonIndexDatasetMapProviderV2.get_all_train_cameras()
    • get_available_subset_names()
    • LlffDatasetMapProvider
      • LlffDatasetMapProvider.downscale_factor
    • RenderedMeshDatasetMapProvider
      • RenderedMeshDatasetMapProvider.num_views
      • RenderedMeshDatasetMapProvider.data_file
      • RenderedMeshDatasetMapProvider.azimuth_range
      • RenderedMeshDatasetMapProvider.distance
      • RenderedMeshDatasetMapProvider.resolution
      • RenderedMeshDatasetMapProvider.use_point_light
      • RenderedMeshDatasetMapProvider.gpu_idx
      • RenderedMeshDatasetMapProvider.path_manager_factory
      • RenderedMeshDatasetMapProvider.path_manager_factory_class_type
      • RenderedMeshDatasetMapProvider.get_dataset_map()
      • RenderedMeshDatasetMapProvider.get_all_train_cameras()
  • pytorch3d.implicitron.evaluation
  • pytorch3d.implicitron.tools
    • jitter_extrinsics()
    • apply_camera_alignment()
    • get_min_max_depth_bounds()
    • volumetric_camera_overlaps()
    • get_rotation_to_best_fit_xy()
    • Circle2D
      • Circle2D.center
      • Circle2D.radius
      • Circle2D.generated_points
    • fit_circle_in_2d()
    • Circle3D
      • Circle3D.center
      • Circle3D.radius
      • Circle3D.normal
      • Circle3D.generated_points
    • fit_circle_in_3d()
    • ReplaceableBase
      • ReplaceableBase.__new__()
    • Configurable
      • Configurable.__new__()
    • run_auto_creation()
    • get_default_args()
    • enable_get_default_args()
    • expand_args_fields()
    • get_default_args_field()
    • remove_unused_components()
    • generate_eval_video_cameras()
    • mask_background()
    • eval_depth()
    • estimate_depth_scale_factor()
    • calc_psnr()
    • calc_mse()
    • calc_bce()
    • binary_cross_entropy_lerp()
    • log_lerp()
    • rgb_l1()
    • huber()
    • neg_iou_loss()
    • safe_sqrt()
    • iou()
    • beta_prior()
    • load_stats()
    • get_model_path()
    • get_optimizer_path()
    • get_stats_path()
    • safe_save_model()
    • save_model()
    • load_model()
    • parse_epoch_from_model_path()
    • get_checkpoint()
    • find_last_checkpoint()
    • purge_epoch()
    • get_rgbd_point_cloud()
    • render_point_cloud_pytorch3d()
    • rasterize_sparse_ray_bundle()
    • rasterize_mc_samples()
    • get_visdom_env()
    • get_visdom_connection()
    • visualize_basics()
    • make_depth_image()