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.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