Version 0.1.6

Yarroudh · Jun 30, 2023 · d341adb · d341adb
1 parent b0c079e
commit d341adb
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -6,4 +6,4 @@ supervision==0.8.0
 segment-geospatial==0.8.1
 rasterio==1.3.7
 laspy==2.4.1
-laszip==0.2.3
+laszip==0.2.3
diff --git a/segment_lidar/samlidar.py b/segment_lidar/samlidar.py
@@ -387,9 +387,10 @@ def write(self, points: np.ndarray, segment_ids: np.ndarray, non_ground: np.ndar
 
         print(f'Writing the segmented point cloud to {save_path}...')
 
+        header = laspy.LasHeader(point_format=3, version="1.3")
+        lidar = laspy.LasData(header=header)
+
         if ground is not None:
-            header = laspy.LasHeader(point_format=3, version="1.3")
-            lidar = laspy.LasData(header=header)
             indices = np.concatenate((non_ground, ground))
             lidar.xyz = points[indices]
             segment_ids = np.append(segment_ids, np.full(len(ground), -1))

diff --git a/setup.py b/setup.py
@@ -18,7 +18,7 @@ def run(self):
 
 setup(
     name="segment-lidar",
-    version='0.1.5',
+    version='0.1.6',
     description="A package for segmenting LiDAR data using Segment-Anything Model (SAM) from Meta AI Research.",
     long_description=open('README.md', encoding='utf-8').read(),
     long_description_content_type='text/markdown',