def _augment_batch_data_level10(self, batch_data):
jittered_data = utils.shift_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