fix bug reported in #3

edwardzhou130 · Apr 20, 2021 · 3a72f23 · 3a72f23
1 parent 8b4a60b
commit 3a72f23
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/dataloader/instance_augmentation.py b/dataloader/instance_augmentation.py
@@ -159,7 +159,10 @@ def instance_flip(self, points,axis,center,flip_type = 1):
 
  def check_occlusion(self,points,center,min_dist=2):
  'check if close to a point'
- dist = np.linalg.norm(points-center,axis=0)
+ if points.ndim == 1:
+ dist = np.linalg.norm(points[np.newaxis,:]-center,axis=1)
+ else:
+ dist = np.linalg.norm(points-center,axis=1)
  return np.all(dist>min_dist)
 
  def rotate_origin(self,xyz,radians):