Update README.md

Yarroudh · Dec 7, 2023 · ca65052 · ca65052
1 parent c4dbf14
commit ca65052
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -52,7 +52,7 @@ viewpoint = view.TopView()
 
 model = samlidar.SamLidar(ckpt_path="sam_vit_h_4b8939.pth")
 points = model.read("pointcloud.las")
-labels, *_ = model.segment(points=points, view=view, image_path="raster.tif", labels_path="labeled.tif")
+labels, *_ = model.segment(points=points, view=viewpoint, image_path="raster.tif", labels_path="labeled.tif")
 model.write(points=points, segment_ids=labels, save_path="segmented.las")
 ```
 
@@ -66,7 +66,7 @@ viewpoint = view.TopView()
 model = samlidar.SamLidar(ckpt_path="sam_vit_h_4b8939.pth")
 points = model.read("pointcloud.las")
 cloud, non_ground, ground = model.csf(points)
-labels, *_ = model.segment(points=cloud, view=view, image_path="raster.tif", labels_path="labeled.tif")
+labels, *_ = model.segment(points=cloud, view=viewpoint, image_path="raster.tif", labels_path="labeled.tif")
 model.write(points=points, non_ground=non_ground, ground=ground, segment_ids=labels, save_path="segmented.las")
 ```