Пример #1
0
    def __init__(self, cfg=None, **kwargs):
        self.remove_environment = cfg.remove_environment
        self.shuffle_points = cfg.shuffle_points
        self.remove_unknown = cfg.remove_unknown_examples
        self.min_points_in_gt = cfg.get("min_points_in_gt", -1)
        self.add_rgb_to_points = cfg.get("add_rgb_to_points", False)
        self.reference_detections = cfg.get("reference_detections", None)
        self.remove_outside_points = cfg.get("remove_outside_points", False)
        self.random_crop = cfg.get("random_crop", False)

        self.normalize_intensity = cfg.get("normalize_intensity", False)

        self.mode = cfg.mode
        if self.mode == "train":
            self.gt_rotation_noise = cfg.gt_rot_noise
            self.gt_loc_noise_std = cfg.gt_loc_noise
            self.global_rotation_noise = cfg.global_rot_noise
            self.global_scaling_noise = cfg.global_scale_noise
            self.global_random_rot_range = cfg.global_rot_per_obj_range
            self.global_translate_noise_std = cfg.global_trans_noise
            self.gt_points_drop = (cfg.gt_drop_percentage, )
            self.remove_points_after_sample = cfg.remove_points_after_sample
            self.class_names = cfg.class_names
            if cfg.db_sampler != None:
                self.db_sampler = build_dbsampler(cfg.db_sampler)
            else:
                self.db_sampler = None

            self.flip_single = cfg.get("flip_single", False)
            self.npoints = cfg.get("npoints", -1)
            self.random_select = cfg.get("random_select", False)

        self.symmetry_intensity = cfg.get("symmetry_intensity", False)
        self.kitti_double = cfg.get("kitti_double", False)
Пример #2
0
    def __init__(self, cfg=None, **kwargs):

        self.shuffle_points = cfg.shuffle_points  # True
        self.remove_environment = cfg.remove_environment  # False
        self.remove_unknown = cfg.remove_unknown_examples  # False

        self.min_points_in_gt = cfg.get("min_points_in_gt", -1)
        self.add_rgb_to_points = cfg.get("add_rgb_to_points", False)
        self.reference_detections = cfg.get("reference_detections", None)
        self.remove_outside_points = cfg.get("remove_outside_points", False)
        self.random_crop = cfg.get("random_crop", False)

        self.mode = cfg.mode
        if self.mode == "train":
            self.gt_loc_noise_std = cfg.gt_loc_noise  # [1.0, 1.0, 0.5],
            self.gt_rotation_noise = cfg.gt_rot_noise  # [-0.785, 0.785],
            self.global_rotation_noise = cfg.global_rot_noise  # [-0.785, 0.785]
            self.global_scaling_noise = cfg.global_scale_noise  # [0.95, 1.05]
            self.global_random_rot_range = cfg.global_rot_per_obj_range  # [0, 0]
            self.global_translate_noise_std = cfg.global_trans_noise  # [0.0, 0.0, 0.0]
            self.gt_points_drop = (cfg.gt_drop_percentage, )  # 0.0
            self.remove_points_after_sample = cfg.remove_points_after_sample  # True
            self.class_names = cfg.class_names  # 'Car'
            self.db_sampler = build_dbsampler(cfg.db_sampler)

            self.npoints = cfg.get("npoints", -1)
            self.random_select = cfg.get("random_select", False)

        self.symmetry_intensity = cfg.get("symmetry_intensity", False)
Пример #3
0
    def __init__(self, cfg=None, **kwargs):
        self.remove_environment = cfg.remove_environment
        self.shuffle_points = cfg.shuffle_points
        self.remove_unknown = cfg.remove_unknown_examples
        self.min_points_in_gt = cfg.get("min_points_in_gt", -1)
        self.add_rgb_to_points = cfg.get("add_rgb_to_points", False)
        self.reference_detections = cfg.get("reference_detections", None)
        self.remove_outside_points = cfg.get("remove_outside_points", False)
        self.random_crop = cfg.get("random_crop", False)
        self.time_stamp_as_last_feature = cfg.get("time_stamp_as_last_feature",
                                                  True)
        self.num_point_features_sampler = cfg.get("num_point_features_sampler",
                                                  5)
        self.num_point_features = cfg.get("num_point_features", 5)
        self.mode = cfg.mode
        if self.mode == "train":
            self.gt_rotation_noise = cfg.gt_rot_noise
            self.gt_loc_noise_std = cfg.gt_loc_noise
            self.global_rotation_noise = cfg.global_rot_noise
            self.global_scaling_noise = cfg.global_scale_noise
            self.global_random_rot_range = cfg.global_rot_per_obj_range
            self.global_translate_noise_std = cfg.global_trans_noise
            self.gt_points_drop = (cfg.gt_drop_percentage, )
            self.remove_points_after_sample = cfg.remove_points_after_sample
            self.class_names = cfg.class_names
            self.db_sampler = build_dbsampler(cfg.db_sampler)
            self.npoints = cfg.get("npoints", -1)
            self.random_select = cfg.get("random_select", False)

        self.symmetry_intensity = cfg.get("symmetry_intensity", False)
Пример #4
0
    def __init__(self, cfg=None, **kwargs):
        self.shuffle_points = cfg.shuffle_points
        self.min_points_in_gt = cfg.get("min_points_in_gt", -1)

        self.mode = cfg.mode
        if self.mode == "train":
            self.global_rotation_noise = cfg.global_rot_noise
            self.global_scaling_noise = cfg.global_scale_noise
            self.class_names = cfg.class_names
            if cfg.db_sampler != None:
                self.db_sampler = build_dbsampler(cfg.db_sampler)
            else:
                self.db_sampler = None

            self.npoints = cfg.get("npoints", -1)

        self.no_augmentation = cfg.get('no_augmentation', False)