API Documentation
- pytorch3d.structures
join_meshes_as_batch()join_meshes_as_scene()MeshesMeshes.__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()PointcloudsPointclouds.__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()VolumesVolumes.__init__()Volumes.__getitem__()Volumes.features()Volumes.densities()Volumes.densities_list()Volumes.features_list()Volumes.get_align_corners()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
- pytorch3d.loss
- pytorch3d.ops
ball_query()corresponding_cameras_alignment()cubify()GraphConvinterpolate_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()SubdivideMeshesconvert_pointclouds_to_tensor()eyes()get_point_covariances()is_pointclouds()wmean()vert_align()
- pytorch3d.renderer
- pytorch3d.renderer.blending
- pytorch3d.renderer.camera_conversions
- pytorch3d.renderer.camera_utils
- pytorch3d.renderer.cameras
CamerasBaseCamerasBase.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()FoVPerspectiveCamerasOpenGLOrthographicCameras()FoVOrthographicCamerasSfMPerspectiveCameras()PerspectiveCamerasSfMOrthographicCameras()OrthographicCamerasget_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
- pytorch3d.renderer.fisheyecameras
- pytorch3d.renderer.lighting
- pytorch3d.renderer.materials
- pytorch3d.renderer.splatter_blend
- pytorch3d.renderer.utils
- pytorch3d.renderer.implicit
- pytorch3d.renderer.implicit.harmonic_embedding
- pytorch3d.renderer.implicit.raymarching
- pytorch3d.renderer.implicit.raysampling
- pytorch3d.renderer.implicit.renderer
- pytorch3d.renderer.implicit.sample_pdf
- pytorch3d.renderer.implicit.utils
- pytorch3d.renderer.mesh
- pytorch3d.renderer.mesh.clip
- pytorch3d.renderer.mesh.rasterize_meshes
- pytorch3d.renderer.mesh.rasterizer
FragmentsRasterizationSettingsRasterizationSettings.image_sizeRasterizationSettings.blur_radiusRasterizationSettings.faces_per_pixelRasterizationSettings.bin_sizeRasterizationSettings.max_faces_openglRasterizationSettings.max_faces_per_binRasterizationSettings.perspective_correctRasterizationSettings.clip_barycentric_coordsRasterizationSettings.cull_backfacesRasterizationSettings.z_clip_valueRasterizationSettings.cull_to_frustum
MeshRasterizer
- pytorch3d.renderer.mesh.renderer
- pytorch3d.renderer.mesh.shader
- pytorch3d.renderer.mesh.shading
- pytorch3d.renderer.mesh.textures
TexturesBaseTextures()TexturesAtlasTexturesAtlas.__init__()TexturesAtlas.clone()TexturesAtlas.detach()TexturesAtlas.atlas_padded()TexturesAtlas.atlas_list()TexturesAtlas.atlas_packed()TexturesAtlas.extend()TexturesAtlas.sample_textures()TexturesAtlas.submeshes()TexturesAtlas.faces_verts_textures_packed()TexturesAtlas.join_batch()TexturesAtlas.join_scene()TexturesAtlas.check_shapes()
TexturesUVTexturesUV.__init__()TexturesUV.clone()TexturesUV.detach()TexturesUV.faces_uvs_padded()TexturesUV.faces_uvs_list()TexturesUV.verts_uvs_padded()TexturesUV.verts_uvs_list()TexturesUV.maps_ids_padded()TexturesUV.maps_ids_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()TexturesUV.submeshes()
TexturesVertexTexturesVertex.__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
- pytorch3d.renderer.opengl
- pytorch3d.renderer.points
- pytorch3d.renderer.points.compositor
- pytorch3d.renderer.points.rasterize_points
- pytorch3d.renderer.points.rasterizer
- pytorch3d.renderer.points.renderer
- pytorch3d.renderer.points.pulsar
- 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()RotateRotateAxisAngleScaleTransform3dTransform3d.__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
- pytorch3d.utils
- pytorch3d.datasets
- pytorch3d.common
- pytorch3d.vis
- pytorch3d.implicitron
- pytorch3d.implicitron.models
- pytorch3d.implicitron.models.base_model
- pytorch3d.implicitron.models.generic_model
GenericModelGenericModel.mask_imagesGenericModel.mask_depthsGenericModel.render_image_widthGenericModel.render_image_heightGenericModel.mask_thresholdGenericModel.output_rasterized_mcGenericModel.bg_colorGenericModel.num_passesGenericModel.chunk_size_gridGenericModel.render_features_dimensionsGenericModel.tqdm_trigger_thresholdGenericModel.n_train_target_viewsGenericModel.sampling_mode_trainingGenericModel.sampling_mode_evaluationGenericModel.global_encoder_class_typeGenericModel.global_encoderGenericModel.raysampler_class_typeGenericModel.raysamplerGenericModel.renderer_class_typeGenericModel.rendererGenericModel.image_feature_extractorGenericModel.image_feature_extractor_class_typeGenericModel.view_pooler_enabledGenericModel.view_poolerGenericModel.implicit_function_class_typeGenericModel.implicit_functionGenericModel.view_metricsGenericModel.view_metrics_class_typeGenericModel.regularization_metricsGenericModel.regularization_metrics_class_typeGenericModel.loss_weightsGenericModel.log_varsGenericModel.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
- pytorch3d.implicitron.models.model_dbir
- pytorch3d.implicitron.models.feature_extractor
- pytorch3d.implicitron.models.feature_extractor.feature_extractor
- pytorch3d.implicitron.models.feature_extractor.resnet_feature_extractor
ResNetFeatureExtractorResNetFeatureExtractor.nameResNetFeatureExtractor.pretrainedResNetFeatureExtractor.stagesResNetFeatureExtractor.normalize_imageResNetFeatureExtractor.image_rescaleResNetFeatureExtractor.first_max_poolResNetFeatureExtractor.proj_dimResNetFeatureExtractor.l2_normResNetFeatureExtractor.add_masksResNetFeatureExtractor.add_imagesResNetFeatureExtractor.global_average_poolResNetFeatureExtractor.feature_rescaleResNetFeatureExtractor.get_feat_dims()ResNetFeatureExtractor.forward()
- pytorch3d.implicitron.models.global_encoder
- pytorch3d.implicitron.models.implicit_function
- pytorch3d.implicitron.models.implicit_function.base
- pytorch3d.implicitron.models.implicit_function.decoding_functions
DecoderActivationDecoderFunctionBaseElementwiseDecoderMLPWithInputSkipsMLPWithInputSkips.n_layersMLPWithInputSkips.input_dimMLPWithInputSkips.output_dimMLPWithInputSkips.skip_dimMLPWithInputSkips.hidden_dimMLPWithInputSkips.input_skipsMLPWithInputSkips.skip_affine_transMLPWithInputSkips.last_layer_bias_initMLPWithInputSkips.last_activationMLPWithInputSkips.use_xavier_initMLPWithInputSkips.forward()
MLPDecoderTransformerWithInputSkipsTransformerEncoderLayer
- pytorch3d.implicitron.models.implicit_function.idr_feature_field
IdrFeatureFieldIdrFeatureField.feature_vector_sizeIdrFeatureField.d_inIdrFeatureField.d_outIdrFeatureField.dimsIdrFeatureField.geometric_initIdrFeatureField.biasIdrFeatureField.skip_inIdrFeatureField.weight_normIdrFeatureField.n_harmonic_functions_xyzIdrFeatureField.pooled_feature_dimIdrFeatureField.encoding_dimIdrFeatureField.forward()
- pytorch3d.implicitron.models.implicit_function.neural_radiance_field
NeuralRadianceFieldBaseNeuralRadianceFieldBase.n_harmonic_functions_xyzNeuralRadianceFieldBase.n_harmonic_functions_dirNeuralRadianceFieldBase.n_hidden_neurons_dirNeuralRadianceFieldBase.latent_dimNeuralRadianceFieldBase.input_xyzNeuralRadianceFieldBase.xyz_ray_dir_in_camera_coordsNeuralRadianceFieldBase.color_dimNeuralRadianceFieldBase.use_integrated_positional_encodingNeuralRadianceFieldBase.get_xyz_embedding_dim()NeuralRadianceFieldBase.allows_multiple_passes()NeuralRadianceFieldBase.forward()
NeuralRadianceFieldImplicitFunctionNeRFormerImplicitFunction
- pytorch3d.implicitron.models.implicit_function.scene_representation_networks
SRNRaymarchFunctionSRNRaymarchFunction.n_harmonic_functionsSRNRaymarchFunction.n_hidden_unitsSRNRaymarchFunction.n_layersSRNRaymarchFunction.in_featuresSRNRaymarchFunction.out_featuresSRNRaymarchFunction.latent_dimSRNRaymarchFunction.xyz_in_camera_coordsSRNRaymarchFunction.raymarch_functionSRNRaymarchFunction.forward()
SRNPixelGeneratorSRNRaymarchHyperNetSRNRaymarchHyperNet.n_harmonic_functionsSRNRaymarchHyperNet.n_hidden_unitsSRNRaymarchHyperNet.n_layersSRNRaymarchHyperNet.n_hidden_units_hypernetSRNRaymarchHyperNet.n_layers_hypernetSRNRaymarchHyperNet.in_featuresSRNRaymarchHyperNet.out_featuresSRNRaymarchHyperNet.latent_dim_hypernetSRNRaymarchHyperNet.latent_dimSRNRaymarchHyperNet.xyz_in_camera_coordsSRNRaymarchHyperNet.forward()
SRNImplicitFunctionSRNHyperNetImplicitFunctionSRNHyperNetImplicitFunction.latent_dim_hypernetSRNHyperNetImplicitFunction.latent_dimSRNHyperNetImplicitFunction.hypernetSRNHyperNetImplicitFunction.pixel_generatorSRNHyperNetImplicitFunction.create_hypernet()SRNHyperNetImplicitFunction.hypernet_tweak_args()SRNHyperNetImplicitFunction.forward()SRNHyperNetImplicitFunction.on_bind_args()
- pytorch3d.implicitron.models.implicit_function.utils
- pytorch3d.implicitron.models.implicit_function.voxel_grid
VoxelGridValuesBaseVoxelGridBaseVoxelGridBase.align_cornersVoxelGridBase.paddingVoxelGridBase.modeVoxelGridBase.n_featuresVoxelGridBase.resolution_changesVoxelGridBase.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()
FullResolutionVoxelGridValuesFullResolutionVoxelGridCPFactorizedVoxelGridValuesCPFactorizedVoxelGridVMFactorizedVoxelGridValuesVMFactorizedVoxelGridValues.vector_components_xVMFactorizedVoxelGridValues.vector_components_yVMFactorizedVoxelGridValues.vector_components_zVMFactorizedVoxelGridValues.matrix_components_xyVMFactorizedVoxelGridValues.matrix_components_yzVMFactorizedVoxelGridValues.matrix_components_xzVMFactorizedVoxelGridValues.basis_matrix
VMFactorizedVoxelGridVoxelGridModuleVoxelGridModule.voxel_grid_class_typeVoxelGridModule.voxel_gridVoxelGridModule.extentsVoxelGridModule.translationVoxelGridModule.init_stdVoxelGridModule.init_meanVoxelGridModule.hold_voxel_grid_as_parametersVoxelGridModule.param_groupsVoxelGridModule.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
VoxelGridImplicitFunctionVoxelGridImplicitFunction.voxel_grid_densityVoxelGridImplicitFunction.voxel_grid_colorVoxelGridImplicitFunction.harmonic_embedder_xyz_density_argsVoxelGridImplicitFunction.harmonic_embedder_xyz_color_argsVoxelGridImplicitFunction.harmonic_embedder_dir_color_argsVoxelGridImplicitFunction.decoder_density_class_typeVoxelGridImplicitFunction.decoder_densityVoxelGridImplicitFunction.decoder_color_class_typeVoxelGridImplicitFunction.decoder_colorVoxelGridImplicitFunction.use_multiple_streamsVoxelGridImplicitFunction.xyz_ray_dir_in_camera_coordsVoxelGridImplicitFunction.scaffold_calculating_epochsVoxelGridImplicitFunction.scaffold_resolutionVoxelGridImplicitFunction.scaffold_empty_space_thresholdVoxelGridImplicitFunction.scaffold_occupancy_chunk_sizeVoxelGridImplicitFunction.scaffold_max_pool_kernel_sizeVoxelGridImplicitFunction.scaffold_filter_pointsVoxelGridImplicitFunction.volume_cropping_epochsVoxelGridImplicitFunction.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
EvaluationModeRenderSamplingModeImplicitronRayBundleImplicitronRayBundle.originsImplicitronRayBundle.directionsImplicitronRayBundle.lengthsImplicitronRayBundle.xysImplicitronRayBundle.camera_idsImplicitronRayBundle.camera_countsImplicitronRayBundle.binsImplicitronRayBundle.pixel_radii_2dImplicitronRayBundle.lengthsImplicitronRayBundle.float_()ImplicitronRayBundle.is_packed()ImplicitronRayBundle.get_padded_xys()
RendererOutputImplicitFunctionWrapperBaseRenderercompute_3d_diagonal_covariance_gaussian()approximate_conical_frustum_as_gaussians()conical_frustum_to_gaussian()
- pytorch3d.implicitron.models.renderer.lstm_renderer
- pytorch3d.implicitron.models.renderer.multipass_ea
MultiPassEmissionAbsorptionRendererMultiPassEmissionAbsorptionRenderer.raymarcher_class_typeMultiPassEmissionAbsorptionRenderer.raymarcherMultiPassEmissionAbsorptionRenderer.n_pts_per_ray_fine_trainingMultiPassEmissionAbsorptionRenderer.n_pts_per_ray_fine_evaluationMultiPassEmissionAbsorptionRenderer.stratified_sampling_coarse_trainingMultiPassEmissionAbsorptionRenderer.stratified_sampling_coarse_evaluationMultiPassEmissionAbsorptionRenderer.append_coarse_samples_to_fineMultiPassEmissionAbsorptionRenderer.density_noise_std_trainMultiPassEmissionAbsorptionRenderer.return_weightsMultiPassEmissionAbsorptionRenderer.blurpool_weightsMultiPassEmissionAbsorptionRenderer.sample_pdf_epsMultiPassEmissionAbsorptionRenderer.forward()
- pytorch3d.implicitron.models.renderer.ray_point_refiner
- pytorch3d.implicitron.models.renderer.ray_sampler
RaySamplerBaseAbstractMaskRaySamplerAbstractMaskRaySampler.image_widthAbstractMaskRaySampler.image_heightAbstractMaskRaySampler.sampling_mode_trainingAbstractMaskRaySampler.sampling_mode_evaluationAbstractMaskRaySampler.n_pts_per_ray_trainingAbstractMaskRaySampler.n_pts_per_ray_evaluationAbstractMaskRaySampler.n_rays_per_image_sampled_from_maskAbstractMaskRaySampler.n_rays_total_trainingAbstractMaskRaySampler.stratified_point_sampling_trainingAbstractMaskRaySampler.stratified_point_sampling_evaluationAbstractMaskRaySampler.cast_ray_bundle_as_coneAbstractMaskRaySampler.forward()
AdaptiveRaySamplerNearFarRaySamplercompute_radii()
- pytorch3d.implicitron.models.renderer.ray_tracing
RayTracingRayTracing.object_bounding_sphereRayTracing.sdf_thresholdRayTracing.line_search_stepRayTracing.line_step_itersRayTracing.sphere_tracing_itersRayTracing.n_stepsRayTracing.n_secant_stepsRayTracing.forward()RayTracing.sphere_tracing()RayTracing.ray_sampler()RayTracing.secant()RayTracing.minimal_sdf_points()
- pytorch3d.implicitron.models.renderer.raymarcher
RaymarcherBaseAccumulativeRaymarcherBaseAccumulativeRaymarcherBase.surface_thicknessAccumulativeRaymarcherBase.bg_colorAccumulativeRaymarcherBase.replicate_last_intervalAccumulativeRaymarcherBase.background_opacityAccumulativeRaymarcherBase.density_reluAccumulativeRaymarcherBase.blend_outputAccumulativeRaymarcherBase.capping_function_typeAccumulativeRaymarcherBase.weight_function_typeAccumulativeRaymarcherBase.__post_init__()AccumulativeRaymarcherBase.forward()
EmissionAbsorptionRaymarcherCumsumRaymarcher
- pytorch3d.implicitron.models.renderer.rgb_net
- pytorch3d.implicitron.models.renderer.sdf_renderer
SignedDistanceFunctionRendererSignedDistanceFunctionRenderer.render_features_dimensionsSignedDistanceFunctionRenderer.object_bounding_sphereSignedDistanceFunctionRenderer.ray_tracerSignedDistanceFunctionRenderer.ray_normal_coloring_network_argsSignedDistanceFunctionRenderer.bg_colorSignedDistanceFunctionRenderer.soft_mask_alphaSignedDistanceFunctionRenderer.ray_tracer_tweak_args()SignedDistanceFunctionRenderer.create_ray_tracer()SignedDistanceFunctionRenderer.requires_object_mask()SignedDistanceFunctionRenderer.forward()
- pytorch3d.implicitron.models.renderer.base
- pytorch3d.implicitron.models.view_pooler
- pytorch3d.implicitron.models.view_pooler.feature_aggregator
ReductionFunctionFeatureAggregatorBaseIdentityFeatureAggregatorReductionFeatureAggregatorAngleWeightedReductionFeatureAggregatorAngleWeightedReductionFeatureAggregator.reduction_functionsAngleWeightedReductionFeatureAggregator.weight_by_ray_angle_gammaAngleWeightedReductionFeatureAggregator.min_ray_angle_weightAngleWeightedReductionFeatureAggregator.get_aggregated_feature_dim()AngleWeightedReductionFeatureAggregator.forward()
AngleWeightedIdentityFeatureAggregator
- pytorch3d.implicitron.models.view_pooler.view_pooler
- pytorch3d.implicitron.models.view_pooler.view_sampler
- pytorch3d.implicitron.models.view_pooler.feature_aggregator
- pytorch3d.implicitron.models.visualization
- pytorch3d.implicitron.dataset in general
- pytorch3d.implicitron.dataset specific datasets
- 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()Circle2Dfit_circle_in_2d()Circle3Dfit_circle_in_3d()ReplaceableBaseConfigurablerun_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()save_stats()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()
- pytorch3d.implicitron.models