Spaces:

jcenaa
/

Segment-Any-RGBD

Runtime error

Jingkang Yang commited on Apr 25, 2023

Commit

97e7ba4

1 Parent(s): 12193a7

update: support scannet

Files changed (1) hide show

app.py CHANGED Viewed

@@ -169,8 +169,6 @@ def greet_scannet(rgb_input, depth_map_input, class_candidates):
     cfg = setup_cfg(args)
     demo = VisualizationDemoIndoor(cfg)
-    from datasets.scannet_preprocess.meta_data.scannet200_constants import CLASS_LABELS_200
-    class_names = list(CLASS_LABELS_200)
     """ args.input = glob.glob(os.path.expanduser(args.input[0]))
     assert args.input, "The input path(s) was not found" """
     start_time = time.time()

     cfg = setup_cfg(args)
     demo = VisualizationDemoIndoor(cfg)
     """ args.input = glob.glob(os.path.expanduser(args.input[0]))
     assert args.input, "The input path(s) was not found" """
     start_time = time.time()