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.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
DatasetBaseDatasetMapDatasetMapProviderBasePathManagerFactoryDataLoaderMapDataLoaderMapProviderBaseSimpleDataLoaderMapProviderDoublePoolBatchSamplerBatchConditioningTypeSequenceDataLoaderMapProviderSequenceDataLoaderMapProvider.batch_sizeSequenceDataLoaderMapProvider.num_workersSequenceDataLoaderMapProvider.dataset_length_trainSequenceDataLoaderMapProvider.dataset_length_valSequenceDataLoaderMapProvider.dataset_length_testSequenceDataLoaderMapProvider.train_conditioning_typeSequenceDataLoaderMapProvider.val_conditioning_typeSequenceDataLoaderMapProvider.test_conditioning_typeSequenceDataLoaderMapProvider.images_per_seq_optionsSequenceDataLoaderMapProvider.sample_consecutive_framesSequenceDataLoaderMapProvider.consecutive_frames_max_gapSequenceDataLoaderMapProvider.consecutive_frames_max_gap_secondsSequenceDataLoaderMapProvider.get_data_loader_map()
DataSourceBaseImplicitronDataSourceImplicitronDataSource.dataset_map_providerImplicitronDataSource.dataset_map_provider_class_typeImplicitronDataSource.data_loader_map_providerImplicitronDataSource.data_loader_map_provider_class_typeImplicitronDataSource.pre_expand()ImplicitronDataSource.get_datasets_and_dataloaders()ImplicitronDataSource.all_train_cameras
SceneBatchSamplerSceneBatchSampler.datasetSceneBatchSampler.batch_sizeSceneBatchSampler.num_batchesSceneBatchSampler.images_per_seq_optionsSceneBatchSampler.sample_consecutive_framesSceneBatchSampler.consecutive_frames_max_gapSceneBatchSampler.consecutive_frames_max_gap_secondsSceneBatchSampler.category_awareSceneBatchSampler.seq_namesSceneBatchSampler.category_to_sequence_namesSceneBatchSampler.categories
- pytorch3d.implicitron.dataset specific datasets
BlenderDatasetMapProviderJsonIndexDatasetMapProviderJsonIndexDatasetMapProvider.categoryJsonIndexDatasetMapProvider.task_strJsonIndexDatasetMapProvider.dataset_rootJsonIndexDatasetMapProvider.n_frames_per_sequenceJsonIndexDatasetMapProvider.test_on_trainJsonIndexDatasetMapProvider.restrict_sequence_nameJsonIndexDatasetMapProvider.test_restrict_sequence_idJsonIndexDatasetMapProvider.assert_single_seqJsonIndexDatasetMapProvider.only_test_setJsonIndexDatasetMapProvider.datasetJsonIndexDatasetMapProvider.dataset_class_typeJsonIndexDatasetMapProvider.path_manager_factoryJsonIndexDatasetMapProvider.path_manager_factory_class_typeJsonIndexDatasetMapProvider.dataset_tweak_args()JsonIndexDatasetMapProvider.create_dataset()JsonIndexDatasetMapProvider.get_dataset_map()JsonIndexDatasetMapProvider.get_all_train_cameras()
JsonIndexDatasetMapProviderV2JsonIndexDatasetMapProviderV2.categoryJsonIndexDatasetMapProviderV2.subset_nameJsonIndexDatasetMapProviderV2.dataset_rootJsonIndexDatasetMapProviderV2.test_on_trainJsonIndexDatasetMapProviderV2.only_test_setJsonIndexDatasetMapProviderV2.load_eval_batchesJsonIndexDatasetMapProviderV2.num_load_workersJsonIndexDatasetMapProviderV2.n_known_frames_for_testJsonIndexDatasetMapProviderV2.dataset_class_typeJsonIndexDatasetMapProviderV2.datasetJsonIndexDatasetMapProviderV2.path_manager_factoryJsonIndexDatasetMapProviderV2.path_manager_factory_class_typeJsonIndexDatasetMapProviderV2.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()LlffDatasetMapProviderRenderedMeshDatasetMapProviderRenderedMeshDatasetMapProvider.num_viewsRenderedMeshDatasetMapProvider.data_fileRenderedMeshDatasetMapProvider.azimuth_rangeRenderedMeshDatasetMapProvider.distanceRenderedMeshDatasetMapProvider.resolutionRenderedMeshDatasetMapProvider.use_point_lightRenderedMeshDatasetMapProvider.gpu_idxRenderedMeshDatasetMapProvider.path_manager_factoryRenderedMeshDatasetMapProvider.path_manager_factory_class_typeRenderedMeshDatasetMapProvider.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()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()