xinwei89 commited on
Commit
91252a6
1 Parent(s): 05b14d0
Files changed (1) hide show
  1. 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