Spaces:
Runtime error
Runtime error
xinwei89
commited on
Commit
·
91252a6
1
Parent(s):
05b14d0
test meta
Browse files- backend.py +3 -0
backend.py
CHANGED
@@ -80,8 +80,11 @@ def visualize_image(im, mode="BOTH", tree_threshold=0.7, building_threshold=0.7,
|
|
80 |
tree_instances = segment_tree(im, tree_threshold)
|
81 |
building_instances = segment_building(im, building_threshold)
|
82 |
instances = Instances.cat([tree_instances, building_instances])
|
|
|
|
|
83 |
|
84 |
visualizer = Visualizer(im[:, :, ::-1],
|
|
|
85 |
scale=0.5,
|
86 |
instance_mode=color_mode)
|
87 |
|
|
|
80 |
tree_instances = segment_tree(im, tree_threshold)
|
81 |
building_instances = segment_building(im, building_threshold)
|
82 |
instances = Instances.cat([tree_instances, building_instances])
|
83 |
+
|
84 |
+
meta = MetadataCatalog.get("predict")
|
85 |
|
86 |
visualizer = Visualizer(im[:, :, ::-1],
|
87 |
+
metadata=meta,
|
88 |
scale=0.5,
|
89 |
instance_mode=color_mode)
|
90 |
|