Commit 2ba0b4e1 authored by liyinhao's avatar liyinhao
Browse files

change wrong words

parent e90a8be6
...@@ -32,18 +32,16 @@ class IndoorFlipData(object): ...@@ -32,18 +32,16 @@ class IndoorFlipData(object):
gt_bboxes_3d[:, 0] = -1 * gt_bboxes_3d[:, 0] gt_bboxes_3d[:, 0] = -1 * gt_bboxes_3d[:, 0]
if not aligned: if not aligned:
gt_bboxes_3d[:, 6] = np.pi - gt_bboxes_3d[:, 6] gt_bboxes_3d[:, 6] = np.pi - gt_bboxes_3d[:, 6]
results['flip_yz'] = True results['flip_yz'] = True
results['gt_boxes_3d'] = gt_bboxes_3d
if aligned and np.random.random() < self.flip_ratio_xz: if aligned and np.random.random() < self.flip_ratio_xz:
# Flipping along the XZ plane # Flipping along the XZ plane
points[:, 1] = -1 * points[:, 1] points[:, 1] = -1 * points[:, 1]
gt_bboxes_3d[:, 1] = -1 * gt_bboxes_3d[:, 1] gt_bboxes_3d[:, 1] = -1 * gt_bboxes_3d[:, 1]
results['flip_xz'] = True results['flip_xz'] = True
results['gt_bboxes_3d'] = gt_bboxes_3d
results['points'] = points
results['points'] = points
results['gt_bboxes_3d'] = gt_bboxes_3d
return results return results
def __repr__(self): def __repr__(self):
...@@ -194,7 +192,7 @@ class IndoorGlobalRotScale(object): ...@@ -194,7 +192,7 @@ class IndoorGlobalRotScale(object):
if self.rot_range is not None: if self.rot_range is not None:
assert len(self.rot_range) == 2, \ assert len(self.rot_range) == 2, \
f'Except length of rot range =2, ' \ f'Expect length of rot range =2, ' \
f'got {len(self.rot_range)}.' f'got {len(self.rot_range)}.'
rot_angle = np.random.uniform(self.rot_range[0], self.rot_range[1]) rot_angle = np.random.uniform(self.rot_range[0], self.rot_range[1])
rot_mat = self._rotz(rot_angle) rot_mat = self._rotz(rot_angle)
...@@ -208,7 +206,7 @@ class IndoorGlobalRotScale(object): ...@@ -208,7 +206,7 @@ class IndoorGlobalRotScale(object):
if self.scale_range is not None: if self.scale_range is not None:
assert len(self.scale_range) == 2, \ assert len(self.scale_range) == 2, \
f'Except length of scale range =2, ' \ f'Expect length of scale range =2, ' \
f'got {len(self.scale_range)}.' f'got {len(self.scale_range)}.'
# Augment point cloud scale # Augment point cloud scale
scale_ratio = np.random.uniform(self.scale_range[0], scale_ratio = np.random.uniform(self.scale_range[0],
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment