Spaces:
Runtime error
Runtime error
JeffLiang
commited on
Commit
·
da442e3
1
Parent(s):
3669297
update
Browse files
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, [
|
52 |
-
['scene0002_00', 0, [0, -
|
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)
|
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)
|