Update sam2point/dataset.py

sam2point/dataset.py

@@ -37,19 +37,6 @@ def load_KITTI_sample(data_path, close=False):
     point = point - pmin
     pmax = point.max(axis=0)
     point = point / pmax
-    
-    # if close:
-    #     x_min, x_max = 0.3, 0.7
-    #     y_min, y_max = 0.3, 0.8
-
-    #     filter_mask = (point[:, 0] > x_min) & (point[:, 0] < x_max) & (point[:, 1] > y_min) & (point[:, 1] < y_max)
-    #     point = point[filter_mask]
-    #     color = color[filter_mask]
-
-    #     pmin = point.min(axis=0)
-    #     point = point - pmin
-    #     pmax = point.max(axis=0)
-    #     point = point / pmax
 
     return point, color
 
@@ -90,9 +77,4 @@ def load_Semantic3D_sample(data_path, id, sample=False):
     pmax = point.max(axis=0)
     point = point / pmax
 
-    # if sample: #for demo
-    #     indices = np.random.choice(point.shape[0], 600000, replace=False)
-    #     point = point[indices]
-    #     color = color[indices]
-
     return point, color