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
- pytorch3d.loss
- pytorch3d.ops
ball_query()
corresponding_cameras_alignment()
cubify()
GraphConv
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
convert_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
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
OpenGLOrthographicCameras()
FoVOrthographicCameras
SfMPerspectiveCameras()
PerspectiveCameras
SfMOrthographicCameras()
OrthographicCameras
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
- 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
Fragments
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
- pytorch3d.renderer.mesh.renderer
- pytorch3d.renderer.mesh.shader
- pytorch3d.renderer.mesh.shading
- pytorch3d.renderer.mesh.textures
TexturesBase
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
- 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()
Rotate
RotateAxisAngle
Scale
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
- pytorch3d.utils
- pytorch3d.datasets
- pytorch3d.common
- pytorch3d.vis
- pytorch3d.implicitron
- pytorch3d.implicitron.models
- pytorch3d.implicitron.models.base_model
- 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
- 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
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.implicit_function
- pytorch3d.implicitron.models.implicit_function.base
- pytorch3d.implicitron.models.implicit_function.decoding_functions
DecoderActivation
DecoderFunctionBase
ElementwiseDecoder
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
TransformerWithInputSkips
TransformerEncoderLayer
- 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
NeRFormerImplicitFunction
- 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
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
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
- 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
FullResolutionVoxelGrid
CPFactorizedVoxelGridValues
CPFactorizedVoxelGrid
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
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
RenderSamplingMode
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
ImplicitFunctionWrapper
BaseRenderer
compute_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
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
- pytorch3d.implicitron.models.renderer.ray_sampler
RaySamplerBase
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
NearFarRaySampler
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
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
CumsumRaymarcher
- pytorch3d.implicitron.models.renderer.rgb_net
- 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.renderer.base
- pytorch3d.implicitron.models.view_pooler
- pytorch3d.implicitron.models.view_pooler.feature_aggregator
ReductionFunction
FeatureAggregatorBase
IdentityFeatureAggregator
ReductionFeatureAggregator
AngleWeightedReductionFeatureAggregator
AngleWeightedReductionFeatureAggregator.reduction_functions
AngleWeightedReductionFeatureAggregator.weight_by_ray_angle_gamma
AngleWeightedReductionFeatureAggregator.min_ray_angle_weight
AngleWeightedReductionFeatureAggregator.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
DatasetBase
DatasetMap
DatasetMapProviderBase
PathManagerFactory
DataLoaderMap
DataLoaderMapProviderBase
SimpleDataLoaderMapProvider
DoublePoolBatchSampler
BatchConditioningType
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
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
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
fit_circle_in_2d()
Circle3D
fit_circle_in_3d()
ReplaceableBase
Configurable
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()
- pytorch3d.implicitron.models