Function render_rgbd

pointersect/inference/main.py:451–699

Render given rgbd images.

(
        input_rgbd_images: RGBDImage,  # (1, q)
        output_dir: str,
        model_filename: T.Union[str, T.List[str]],
        n_output_imgs: int,
        output_camera_trajectory_mode: str,
        # for pointersect
        k: int,
        # for comparison with baselines
        render_pointersect: bool = True,
        render_npbgpp: bool = False,
        render_surfel: bool = True,
        render_nglod: bool = False,
        render_ngp: bool = False,
        render_dsnerf: bool = False,
        render_ibrnet: bool = False,
        render_poisson: bool = True,
        densify_neural_points: bool = False,
        # other settings
        rnd_seed: int = 0,
        output_camera_trajectory_params: T.Dict[str, T.Any] = None,
        output_camera_setting: T.Dict[str, T.Any] = None,
        save_settings: T.Dict[str, T.Any] = None,
        pr_setting: T.Dict[str, T.Any] = None,
        model_loading_settings: T.Dict[str, T.Any] = None,
        max_ray_chunk_size: int = int(1e4),  # k=40: int(4e4),
        max_pr_chunk_size: int = -1,
        max_model_chunk_size: int = -1,
        th_hit_prob: float = 0.5,
        gt_rgbd_images: T.Optional[RGBDImage] = None,
        input_point_cloud_subsample: int = 1,
        surfel_point_size: float = 1.,
        force_same_intrinsic: bool = False,
        voxel_downsample_cell_width: float = -1,
        voxel_downsample_sigma: float = 0.5,
        neural_point_upsample_ratio_x48: int = 1,
        total_nglod_epoch: int = 50,
        nglod_test_every_iter: int = 50,
        total_ngp_epoch: int = 50,
        ngp_test_every_iter: int = 50,
        total_dsnerf_iter: int = 50000,
        dsnerf_test_every_iter: int = 5000,
        test_plane_normal: bool = False,
        drop_point_cloud_features: bool = False,
        drop_point_cloud_normal: bool = False,
        pcd_outlier_removal_radius: float = -1,
        pcd_outlier_removal_num_points: int = -1,
        ibrnet_chunk_size: int = 4096,
        **kwargs,
)

def render_rgbd(
        input_rgbd_images: RGBDImage,  # (1, q)
        output_dir: str,
        model_filename: T.Union[str, T.List[str]],
        n_output_imgs: int,
        output_camera_trajectory_mode: str,
        # for pointersect
        k: int,
        # for comparison with baselines
        render_pointersect: bool = True,
        render_npbgpp: bool = False,
        render_surfel: bool = True,
        render_nglod: bool = False,
        render_ngp: bool = False,
        render_dsnerf: bool = False,
        render_ibrnet: bool = False,
        render_poisson: bool = True,
        densify_neural_points: bool = False,
        # other settings
        rnd_seed: int = 0,
        output_camera_trajectory_params: T.Dict[str, T.Any] = None,
        output_camera_setting: T.Dict[str, T.Any] = None,
        save_settings: T.Dict[str, T.Any] = None,
        pr_setting: T.Dict[str, T.Any] = None,
        model_loading_settings: T.Dict[str, T.Any] = None,
        max_ray_chunk_size: int = int(1e4),  # k=40: int(4e4),
        max_pr_chunk_size: int = -1,
        max_model_chunk_size: int = -1,
        th_hit_prob: float = 0.5,
        gt_rgbd_images: T.Optional[RGBDImage] = None,
        input_point_cloud_subsample: int = 1,
        surfel_point_size: float = 1.,
        force_same_intrinsic: bool = False,
        voxel_downsample_cell_width: float = -1,
        voxel_downsample_sigma: float = 0.5,
        neural_point_upsample_ratio_x48: int = 1,
        total_nglod_epoch: int = 50,
        nglod_test_every_iter: int = 50,
        total_ngp_epoch: int = 50,
        ngp_test_every_iter: int = 50,
        total_dsnerf_iter: int = 50000,
        dsnerf_test_every_iter: int = 5000,
        test_plane_normal: bool = False,
        drop_point_cloud_features: bool = False,
        drop_point_cloud_normal: bool = False,
        pcd_outlier_removal_radius: float = -1,
        pcd_outlier_removal_num_points: int = -1,
        ibrnet_chunk_size: int = 4096,
        **kwargs,
):
    """Render given rgbd images."""

    setting_dict = get_settings(
        output_camera_trajectory_params=output_camera_trajectory_params,
        output_camera_setting=output_camera_setting,
        save_settings=save_settings,
        pr_setting=pr_setting,
        model_loading_settings=model_loading_settings,