JeffLiang commited on
Commit
da442e3
·
1 Parent(s): 3669297
Files changed (2) hide show
  1. app.py +2 -2
  2. sam_3d.py +1 -1
app.py CHANGED
@@ -48,8 +48,8 @@ for scene_name in ['scene0000_00', 'scene0001_00', 'scene0002_00']:
48
  plydatas.append(data)
49
 
50
  examples = [['scene0000_00', 0, [0, -2.5, 0.7], pc_to_plot(plydatas[0])],
51
- ['scene0001_00', 0, [0, -2.5, 1], pc_to_plot(plydatas[1])],
52
- ['scene0002_00', 0, [0, -2.5, 1], pc_to_plot(plydatas[2])],]
53
 
54
  title = 'Segment_Anything on 3D in-door point clouds'
55
 
 
48
  plydatas.append(data)
49
 
50
  examples = [['scene0000_00', 0, [0, -2.5, 0.7], pc_to_plot(plydatas[0])],
51
+ ['scene0001_00', 0, [1.9, 1.1, 0.5], pc_to_plot(plydatas[1])],
52
+ ['scene0002_00', 0, [-0.7, -0.3, 1.9], pc_to_plot(plydatas[2])],]
53
 
54
  title = 'Segment_Anything on 3D in-door point clouds'
55
 
sam_3d.py CHANGED
@@ -81,7 +81,7 @@ def plot_3d(xdata, ydata, zdata, color=None, b_min=2, b_max=8, view=(45, 45)):
81
 
82
  class SAM3DDemo(object):
83
  def __init__(self, sam_model, sam_ckpt, scene_name):
84
- sam = sam_model_registry[sam_model](checkpoint=sam_ckpt).cuda()
85
  self.predictor = SamPredictor(sam)
86
  self.scene_name = scene_name
87
  scene_path = os.path.join('./scannet_data', scene_name)
 
81
 
82
  class SAM3DDemo(object):
83
  def __init__(self, sam_model, sam_ckpt, scene_name):
84
+ sam = sam_model_registry[sam_model](checkpoint=sam_ckpt)
85
  self.predictor = SamPredictor(sam)
86
  self.scene_name = scene_name
87
  scene_path = os.path.join('./scannet_data', scene_name)