update

app.py

@@ -48,8 +48,8 @@ for scene_name in ['scene0000_00', 'scene0001_00', 'scene0002_00']:
     plydatas.append(data)
 
 examples = [['scene0000_00', 0, [0, -2.5, 0.7], pc_to_plot(plydatas[0])],
-            ['scene0001_00', 0, [0, -2.5, 1], pc_to_plot(plydatas[1])],
-            ['scene0002_00', 0, [0, -2.5, 1], pc_to_plot(plydatas[2])],]
+            ['scene0001_00', 0, [1.9, 1.1, 0.5], pc_to_plot(plydatas[1])],
+            ['scene0002_00', 0, [-0.7, -0.3, 1.9], pc_to_plot(plydatas[2])],]
 
 title = 'Segment_Anything on 3D in-door point clouds'
 

sam_3d.py

@@ -81,7 +81,7 @@ def plot_3d(xdata, ydata, zdata, color=None, b_min=2, b_max=8, view=(45, 45)):
 
 class SAM3DDemo(object):
     def __init__(self, sam_model, sam_ckpt, scene_name):
-        sam = sam_model_registry[sam_model](checkpoint=sam_ckpt).cuda()
+        sam = sam_model_registry[sam_model](checkpoint=sam_ckpt)
         self.predictor = SamPredictor(sam)
         self.scene_name = scene_name
         scene_path = os.path.join('./scannet_data', scene_name)