bug fix

backend.py

@@ -46,7 +46,6 @@ building_predictor = DefaultPredictor(building_cfg)
 # A function that runs the buildings model on an given image and confidence threshold
 def segment_building(im, confidence_threshold):
     building_cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = confidence_threshold
-    im = np.array(im)
     outputs = building_predictor(im)
     building_instances = outputs["instances"].to("cpu")
 
@@ -55,7 +54,6 @@ def segment_building(im, confidence_threshold):
 # A function that runs the trees model on an given image and confidence threshold
 def segment_tree(im, confidence_threshold):
     tree_cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = confidence_threshold
-    im = np.array(im)
     outputs = tree_predictor(im)
     tree_instances = outputs["instances"].to("cpu")
 
@@ -71,6 +69,7 @@ def map_color_mode(color_mode):
         return ColorMode.SEGMENTATION
 
 def visualize_image(im, mode, tree_threshold, building_threshold, color_mode=ColorMode.SEGMENTATION):
+    im = np.array(im)
     color_mode = map_color_mode(color_mode)
 
     if mode == "Trees":