Class MeshDataset

pointersect/data/mesh_dataset_v2.py:19–335

class MeshDataset:

    def __init__(
            self,
            mesh: structures.Mesh,
            n_target_imgs: int = 3,
            n_imgs: int = 3,
            total: int = 10000,
            input_camera_setting: T.Dict[str, T.Any] = None,
            input_camera_trajectory_params: T.Dict[str, T.Any] = None,
            output_camera_setting: T.Dict[str, T.Any] = None,
            output_camera_trajectory_params: T.Dict[str, T.Any] = None,
            rng_seed: T.Union[int, None] = 0,
            render_method: str = 'ray_cast',
            texture_filenames: T.List[str] = None,
            texture_crop_method: str = 'full',  # 'random', number
            min_subsample: int = 1,
            max_subsample: int = 1,
    ):
        """
        Args:
            mesh:
                class Mesh containing cleaned and scaled mesh and ray_casting scene.
            n_target_imgs:
                number of target images for a pcd sample (think of it as random rays)
            n_imgs:
                number of input images to create an input point cloud
            total:
                number of (pcd_sample, target) pairs (dataset size)
            input_camera_settings:
                width_px: number of pixel in the x direction
                height_px: number of pixel in the x direction
                fov:  field ov view in degree
                ray_offsets:  'center', 'random'
            input_camera_trajectory_params:
                mode: mode of the camera trajectory
                min_r: minimum distance from camera to the mesh origin. None: min_r will be set to the min aabb size
                max_r: maximum distance from camera to the mesh origin.
                max_angle: spread of the images in viewing direction (in degree)
                local_max_angle: small local rotation of each images (in degree)
            output_camera_settings:
                see `input_camera_settings`
            output_camera_trajectory_params:
                see `input_camera_trajectory_params`
                ray_perturbation_angle:  small angle to perturb target ray direction (in degree). If None: use fov / width_px.
            rng_seed:
                random seed to use
            render_method:
                'rasterization'
                'ray_cast'
            texture_filenames:
                list of texture images to randomly use to replace the original texture
            texture_crop_method:
                'full': use the full image
                int: random crop
            min_subsample:
                to randomly generate subsampled input rgbd images.
            max_subsample: