def _augment_batch_data_level9(self, batch_data):
     jittered_data = utils.random_scale_point_cloud(batch_data[:, :, 0:3])
     batch_data[:, :, 0:3] = jittered_data
     if self.dropout_ratio > 0:
         batch_data = utils.random_point_dropout(
             batch_data, max_dropout_ratio=self.dropout_ratio)
     return batch_data
 def _augment_batch_data_level7(self, batch_data):
     jittered_data = utils.random_scale_point_cloud(batch_data[:, :, 0:3],
                                                    scale_low=0.7,
                                                    scale_high=1.4)
     jittered_data = utils.shift_point_cloud(jittered_data, shift_range=0.1)
     batch_data[:, :, 0:3] = jittered_data
     if self.dropout_ratio > 0:
         batch_data = utils.random_point_dropout(
             batch_data, max_dropout_ratio=self.dropout_ratio)
     return batch_data
    def _augment_batch_data_level3(self, batch_data):
        if self.normal_channel:
            rotated_data = utils.rotate_point_cloud_with_normal(batch_data)
            rotated_data = utils.rotate_perturbation_point_cloud_with_normal(
                rotated_data)
        else:
            rotated_data = utils.rotate_point_cloud(batch_data)
            rotated_data = utils.rotate_perturbation_point_cloud(rotated_data)

        jittered_data = utils.random_scale_point_cloud(rotated_data[:, :, 0:3])
        jittered_data = utils.shift_point_cloud(jittered_data)
        jittered_data = utils.jitter_point_cloud(jittered_data)
        rotated_data[:, :, 0:3] = jittered_data
        if self.dropout_ratio > 0:
            rotated_data = utils.random_point_dropout(
                rotated_data, max_dropout_ratio=self.dropout_ratio)
        return rotated_data